Skip to content

Commit

Permalink
Print spaces for readability.
Browse files Browse the repository at this point in the history
  • Loading branch information
wujingyue committed Nov 15, 2024
1 parent 4ac5649 commit 0843fc4
Showing 1 changed file with 4 additions and 4 deletions.
8 changes: 4 additions & 4 deletions csrc/preseg_passes/reorder_sharded_axis.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -28,22 +28,22 @@ void ReorderShardedAxisPass::runPass(Fusion* fusion) {
}
NVF_ERROR(
ir_utils::isTvOp(expr),
"Non-tv op is not supported:",
"Non-tv op is not supported: ",
expr->toString());
NVF_ERROR(
expr->outputs().size() == 1,
"Resharding operations can only have one output",
"Resharding operations can only have one output: ",
expr->toString());
NVF_ERROR(
expr->inputs().size() == 1,
"Resharding operations can have only one input",
"Resharding operations can have only one input: ",
expr->toString());
auto* output = expr->outputs().at(0)->as<TensorView>();
auto* input = expr->inputs().at(0)->as<TensorView>();
auto [shard_additions, shard_deletions] = getShardingChanges(expr);
NVF_ERROR(
shard_additions.size() + shard_deletions.size() <= 1,
"Resharding expr can only support one axis:",
"Resharding expr can only support one axis: ",
expr->toString())

// For gather operations i.e. ID goes from sharded to unsharded
Expand Down

0 comments on commit 0843fc4

Please sign in to comment.