We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 0d68a55 commit 39e427aCopy full SHA for 39e427a
csrc/runtime/fusion_kernel_runtime.cpp
@@ -476,11 +476,7 @@ void FusionKernelRuntime::compileFusionParallel(KernelArgumentHolder args) {
476
std::vector<Val*>{out_clone});
477
hic->pushBackTopLevelExprs(launch_kernel);
478
} else {
479
- const bool is_resharding = std::any_of(
480
- group_to_run->exprs().begin(),
481
- group_to_run->exprs().end(),
482
- [](auto expr) { return isResharding(expr); });
483
- if (is_resharding) {
+ if (group_to_run->schedulerType() == SchedulerType::Communication) {
484
auto deviceid = Communicator::getInstance().deviceId();
485
NVF_ERROR(
486
group_to_run->exprs().size() == 1,
0 commit comments