File tree 1 file changed +0
-24
lines changed
1 file changed +0
-24
lines changed Original file line number Diff line number Diff line change @@ -516,30 +516,6 @@ void FusionKernelRuntime::compileFusionParallel(KernelArgumentHolder args) {
516
516
group_to_run->schedulerType ());
517
517
if (group_to_run->schedulerType () == SchedulerType::Communication) {
518
518
// 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
- }
543
519
} else {
544
520
// push back segment's exprs into the container as top level
545
521
// expressions
You can’t perform that action at this time.
0 commit comments