Skip to content

Commit c3a186e

Browse files
committed
remove lowering for future pr
1 parent 2768635 commit c3a186e

File tree

1 file changed

+0
-24
lines changed

1 file changed

+0
-24
lines changed

csrc/runtime/fusion_kernel_runtime.cpp

-24
Original file line numberDiff line numberDiff line change
@@ -516,30 +516,6 @@ void FusionKernelRuntime::compileFusionParallel(KernelArgumentHolder args) {
516516
group_to_run->schedulerType());
517517
if (group_to_run->schedulerType() == SchedulerType::Communication) {
518518
// TODO: Implement communication lowering
519-
auto deviceid = Communicator::getInstance().deviceId();
520-
NVF_ERROR(
521-
group_to_run->exprs().size() == 1,
522-
"Communication segments must contain only one Expr");
523-
HostIrLower lower;
524-
for (auto* expr : lower.lower(
525-
ir_cloner.clone(group_to_run->exprs().at(0)), deviceid)) {
526-
// Allocate the recv buffers of communications
527-
if (expr->isA<Communication>()) {
528-
auto* communication = expr->as<Communication>();
529-
TensorView* tv = communication->out();
530-
if (tv->getDeviceMesh().has(deviceid)) {
531-
auto* allocate =
532-
IrBuilder::create<kir::Allocate>(tv, MemoryType::Global);
533-
hic->pushBackTopLevelExprs(allocate);
534-
}
535-
}
536-
hic->pushBackTopLevelExprs(expr);
537-
if (expr->isA<Communication>()) {
538-
auto wait =
539-
IrBuilder::create<hir::Wait>(expr->as<Communication>());
540-
hic->pushBackTopLevelExprs(wait);
541-
}
542-
}
543519
} else {
544520
// push back segment's exprs into the container as top level
545521
// expressions

0 commit comments

Comments
 (0)