diff --git a/cpp/velox/compute/WholeStageResultIterator.cc b/cpp/velox/compute/WholeStageResultIterator.cc index b0dac1d65fde..83749061c1b8 100644 --- a/cpp/velox/compute/WholeStageResultIterator.cc +++ b/cpp/velox/compute/WholeStageResultIterator.cc @@ -482,9 +482,12 @@ std::unordered_map WholeStageResultIterator::getQueryC // Find offheap size from Spark confs. If found, set the max memory usage of partial aggregation. // FIXME this uses process-wise off-heap memory which is not for task try { - auto defaultTimezone = veloxCfg_->get(kDefaultSessionTimezone, ""); - configs[velox::core::QueryConfig::kSessionTimezone] = - veloxCfg_->get(kSessionTimezone, defaultTimezone); + if (veloxCfg_->isValueExists(kDefaultSessionTimezone)) { + configs[velox::core::QueryConfig::kSessionTimezone] = veloxCfg_->get(kDefaultSessionTimezone, ""); + } + if (veloxCfg_->isValueExists(kSessionTimezone)) { + configs[velox::core::QueryConfig::kSessionTimezone] = veloxCfg_->get(kSessionTimezone, ""); + } // Adjust timestamp according to the above configured session timezone. configs[velox::core::QueryConfig::kAdjustTimestampToTimezone] = "true"; // Align Velox size function with Spark. diff --git a/ep/build-velox/src/get_velox.sh b/ep/build-velox/src/get_velox.sh index 9344b6ef0ae3..f6291d47e4e4 100755 --- a/ep/build-velox/src/get_velox.sh +++ b/ep/build-velox/src/get_velox.sh @@ -17,7 +17,7 @@ set -exu VELOX_REPO=https://github.com/oap-project/velox.git -VELOX_BRANCH=2024_05_01 +VELOX_BRANCH=2024_05_02 VELOX_HOME="" #Set on run gluten on HDFS