Skip to content

Commit

Permalink
fix style
Browse files Browse the repository at this point in the history
  • Loading branch information
zhli1142015 committed Nov 25, 2023
1 parent 4724acf commit a567df8
Show file tree
Hide file tree
Showing 3 changed files with 6 additions and 3 deletions.
3 changes: 2 additions & 1 deletion cpp/core/shuffle/Partitioner.cc
Original file line number Diff line number Diff line change
Expand Up @@ -23,7 +23,8 @@

namespace gluten {

arrow::Result<std::shared_ptr<Partitioner>> Partitioner::make(Partitioning partitioning, int32_t numPartitions, int32_t partitionKeySeed) {
arrow::Result<std::shared_ptr<Partitioner>>
Partitioner::make(Partitioning partitioning, int32_t numPartitions, int32_t partitionKeySeed) {
switch (partitioning) {
case Partitioning::kHash:
return std::make_shared<HashPartitioner>(numPartitions);
Expand Down
3 changes: 2 additions & 1 deletion cpp/core/shuffle/Partitioner.h
Original file line number Diff line number Diff line change
Expand Up @@ -26,7 +26,8 @@ namespace gluten {

class Partitioner {
public:
static arrow::Result<std::shared_ptr<Partitioner>> make(Partitioning partitioning, int32_t numPartitions, int32_t partitionKeySeed);
static arrow::Result<std::shared_ptr<Partitioner>>
make(Partitioning partitioning, int32_t numPartitions, int32_t partitionKeySeed);

// Whether the first column is partition key.
bool hasPid() const {
Expand Down
3 changes: 2 additions & 1 deletion cpp/velox/shuffle/VeloxShuffleWriter.cc
Original file line number Diff line number Diff line change
Expand Up @@ -411,7 +411,8 @@ arrow::Status VeloxShuffleWriter::init() {
VELOX_CHECK_NOT_NULL(options_.memory_pool);

ARROW_ASSIGN_OR_RAISE(partitionWriter_, partitionWriterCreator_->make(this));
ARROW_ASSIGN_OR_RAISE(partitioner_, Partitioner::make(options_.partitioning, numPartitions_, options_.partition_key_seed));
ARROW_ASSIGN_OR_RAISE(
partitioner_, Partitioner::make(options_.partitioning, numPartitions_, options_.partition_key_seed));

// pre-allocated buffer size for each partition, unit is row count
// when partitioner is SinglePart, partial variables don`t need init
Expand Down

0 comments on commit a567df8

Please sign in to comment.