Skip to content

Commit 39e427a

Browse files
committed
use schedulertype
1 parent 0d68a55 commit 39e427a

File tree

1 file changed

+1
-5
lines changed

1 file changed

+1
-5
lines changed

csrc/runtime/fusion_kernel_runtime.cpp

+1-5
Original file line numberDiff line numberDiff line change
@@ -476,11 +476,7 @@ void FusionKernelRuntime::compileFusionParallel(KernelArgumentHolder args) {
476476
std::vector<Val*>{out_clone});
477477
hic->pushBackTopLevelExprs(launch_kernel);
478478
} 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) {
479+
if (group_to_run->schedulerType() == SchedulerType::Communication) {
484480
auto deviceid = Communicator::getInstance().deviceId();
485481
NVF_ERROR(
486482
group_to_run->exprs().size() == 1,

0 commit comments

Comments
 (0)