Skip to content

Instantly share code, notes, and snippets.

@luhenry
Created April 24, 2023 18:09
Show Gist options
  • Save luhenry/7d877a6e7c2cb2ef2779e6afdd7927e9 to your computer and use it in GitHub Desktop.
Save luhenry/7d877a6e7c2cb2ef2779e6afdd7927e9 to your computer and use it in GitHub Desktop.
diff --git a/velox/substrait/SubstraitToVeloxPlan.cpp b/velox/substrait/SubstraitToVeloxPlan.cpp
index 7dd849201..d1560b79b 100644
--- a/velox/substrait/SubstraitToVeloxPlan.cpp
+++ b/velox/substrait/SubstraitToVeloxPlan.cpp
@@ -1739,6 +1739,12 @@ void SubstraitVeloxPlanConverter::setFilterMap(
}
setColInfoMap<int64_t>(functionName, colIdxVal, val, reverse, colInfoMap);
break;
+ case TypeKind::REAL:
+ if (substraitLit) {
+ val = variant(substraitLit.value().fp32());
+ }
+ setColInfoMap<float>(functionName, colIdxVal, val, reverse, colInfoMap);
+ break;
case TypeKind::DOUBLE:
if (substraitLit) {
val = variant(substraitLit.value().fp64());
@@ -2089,6 +2095,10 @@ connector::hive::SubfieldFilters SubstraitVeloxPlanConverter::mapToFilters(
constructSubfieldFilters<TypeKind::BIGINT, common::BigintRange>(
colIdx, inputNameList[colIdx], colInfoMap[colIdx], filters);
break;
+ case TypeKind::REAL:
+ constructSubfieldFilters<TypeKind::REAL, common::Filter>(
+ colIdx, inputNameList[colIdx], colInfoMap[colIdx], filters);
+ break;
case TypeKind::DOUBLE:
constructSubfieldFilters<TypeKind::DOUBLE, common::Filter>(
colIdx, inputNameList[colIdx], colInfoMap[colIdx], filters);
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment