Skip to content

Commit

Permalink
[VL] Daily Update Velox Version (2024_03_07) (apache#4877)
Browse files Browse the repository at this point in the history
Co-authored-by: PHILO-HE <feilong.he@intel.com>
  • Loading branch information
2 people authored and taiyang-li committed Oct 9, 2024
1 parent 1d1e999 commit 4c58d52
Show file tree
Hide file tree
Showing 7 changed files with 6 additions and 15 deletions.
3 changes: 1 addition & 2 deletions cpp/velox/benchmarks/ColumnarToRowBenchmark.cc
Original file line number Diff line number Diff line change
Expand Up @@ -96,8 +96,7 @@ class GoogleBenchmarkColumnarToRow {
ArrowArray arrowArray;
ArrowSchema arrowSchema;
ASSERT_NOT_OK(arrow::ExportRecordBatch(rb, &arrowArray, &arrowSchema));
return velox::importFromArrowAsOwner(
arrowSchema, arrowArray, ArrowUtils::getBridgeOptions(), gluten::defaultLeafVeloxMemoryPool().get());
return velox::importFromArrowAsOwner(arrowSchema, arrowArray, gluten::defaultLeafVeloxMemoryPool().get());
}

protected:
Expand Down
3 changes: 1 addition & 2 deletions cpp/velox/benchmarks/ParquetWriteBenchmark.cc
Original file line number Diff line number Diff line change
Expand Up @@ -102,8 +102,7 @@ class GoogleBenchmarkParquetWrite {
ArrowArray arrowArray;
ArrowSchema arrowSchema;
ASSERT_NOT_OK(arrow::ExportRecordBatch(rb, &arrowArray, &arrowSchema));
auto vp = velox::importFromArrowAsOwner(
arrowSchema, arrowArray, gluten::ArrowUtils::getBridgeOptions(), gluten::defaultLeafVeloxMemoryPool().get());
auto vp = velox::importFromArrowAsOwner(arrowSchema, arrowArray, gluten::defaultLeafVeloxMemoryPool().get());
return std::make_shared<VeloxColumnarBatch>(std::dynamic_pointer_cast<velox::RowVector>(vp));
}

Expand Down
5 changes: 1 addition & 4 deletions cpp/velox/benchmarks/common/BenchmarkUtils.h
Original file line number Diff line number Diff line change
Expand Up @@ -88,10 +88,7 @@ void abortIfFileNotExists(const std::string& filepath);
inline std::shared_ptr<gluten::ColumnarBatch> convertBatch(std::shared_ptr<gluten::ColumnarBatch> cb) {
if (cb->getType() != "velox") {
auto vp = facebook::velox::importFromArrowAsOwner(
*cb->exportArrowSchema(),
*cb->exportArrowArray(),
gluten::ArrowUtils::getBridgeOptions(),
gluten::defaultLeafVeloxMemoryPool().get());
*cb->exportArrowSchema(), *cb->exportArrowArray(), gluten::defaultLeafVeloxMemoryPool().get());
return std::make_shared<gluten::VeloxColumnarBatch>(std::dynamic_pointer_cast<facebook::velox::RowVector>(vp));
} else {
return cb;
Expand Down
2 changes: 0 additions & 2 deletions cpp/velox/compute/WholeStageResultIterator.cc
Original file line number Diff line number Diff line change
Expand Up @@ -506,8 +506,6 @@ std::unordered_map<std::string, std::string> WholeStageResultIterator::getQueryC
configs[velox::core::QueryConfig::kMaxSplitPreloadPerDriver] =
std::to_string(veloxCfg_->get<int32_t>(kVeloxSplitPreloadPerDriver, 2));

configs[velox::core::QueryConfig::kArrowBridgeTimestampUnit] = "6";

// Disable driver cpu time slicing.
configs[velox::core::QueryConfig::kDriverCpuTimeSliceLimitMs] = "0";

Expand Down
3 changes: 1 addition & 2 deletions cpp/velox/memory/VeloxColumnarBatch.cc
Original file line number Diff line number Diff line change
Expand Up @@ -118,8 +118,7 @@ std::shared_ptr<VeloxColumnarBatch> VeloxColumnarBatch::from(
auto compositeVeloxVector = makeRowVector(childNames, childVectors, cb->numRows(), pool);
return std::make_shared<VeloxColumnarBatch>(compositeVeloxVector);
}
auto vp = velox::importFromArrowAsOwner(
*cb->exportArrowSchema(), *cb->exportArrowArray(), ArrowUtils::getBridgeOptions(), pool);
auto vp = velox::importFromArrowAsOwner(*cb->exportArrowSchema(), *cb->exportArrowArray(), pool);
return std::make_shared<VeloxColumnarBatch>(std::dynamic_pointer_cast<velox::RowVector>(vp));
}

Expand Down
3 changes: 1 addition & 2 deletions cpp/velox/utils/VeloxArrowUtils.cc
Original file line number Diff line number Diff line change
Expand Up @@ -50,8 +50,7 @@ arrow::Result<std::shared_ptr<ColumnarBatch>> recordBatch2VeloxColumnarBatch(con
ArrowArray arrowArray;
ArrowSchema arrowSchema;
RETURN_NOT_OK(arrow::ExportRecordBatch(rb, &arrowArray, &arrowSchema));
auto vp = velox::importFromArrowAsOwner(
arrowSchema, arrowArray, ArrowUtils::getBridgeOptions(), gluten::defaultLeafVeloxMemoryPool().get());
auto vp = velox::importFromArrowAsOwner(arrowSchema, arrowArray, gluten::defaultLeafVeloxMemoryPool().get());
return std::make_shared<VeloxColumnarBatch>(std::dynamic_pointer_cast<velox::RowVector>(vp));
}

Expand Down
2 changes: 1 addition & 1 deletion ep/build-velox/src/get_velox.sh
Original file line number Diff line number Diff line change
Expand Up @@ -17,7 +17,7 @@
set -exu

VELOX_REPO=https://github.com/oap-project/velox.git
VELOX_BRANCH=2024_03_06
VELOX_BRANCH=2024_03_07
VELOX_HOME=""

#Set on run gluten on HDFS
Expand Down

0 comments on commit 4c58d52

Please sign in to comment.