@@ -320,7 +320,6 @@ def alltoall_pooled(
320
320
cumsum_dim_sum_per_rank_tensor = cumsum_dim_sum_per_rank_tensor ,
321
321
codecs = codecs ,
322
322
)
323
- # pyre-fixme[16]: `All2All_Pooled_Req` has no attribute `apply`.
324
323
All2All_Pooled_Req .apply (group , myreq , a2ai , a2a_pooled_embs_tensor )
325
324
return myreq
326
325
@@ -386,7 +385,6 @@ def alltoall_sequence(
386
385
)
387
386
# sequence of embeddings, bags are definitely non-uniform
388
387
389
- # pyre-fixme[16]: `All2All_Seq_Req` has no attribute `apply`.
390
388
All2All_Seq_Req .apply (group , myreq , a2ai , a2a_sequence_embs_tensor )
391
389
return myreq
392
390
@@ -448,7 +446,6 @@ def alltoallv(
448
446
codecs = codecs ,
449
447
)
450
448
451
- # pyre-fixme[16]: `All2Allv_Req` has no attribute `apply`.
452
449
All2Allv_Req .apply (group , myreq , a2ai , inputs )
453
450
454
451
return myreq
@@ -486,7 +483,6 @@ def reduce_scatter_pooled(
486
483
rsi = ReduceScatterInfo (
487
484
input_sizes = [tensor .size () for tensor in inputs ], codecs = codecs
488
485
)
489
- # pyre-fixme[16]
490
486
ReduceScatter_Req .apply (group , myreq , rsi , * inputs )
491
487
return myreq
492
488
@@ -520,7 +516,6 @@ def reduce_scatter_base_pooled(
520
516
521
517
myreq = Request (group , device = inputs .device )
522
518
rsi = ReduceScatterBaseInfo (input_sizes = inputs .size (), codecs = codecs )
523
- # pyre-fixme[16]
524
519
ReduceScatterBase_Req .apply (group , myreq , rsi , inputs )
525
520
return myreq
526
521
@@ -554,7 +549,6 @@ def all_gather_base_pooled(
554
549
555
550
myreq = Request (group , device = input .device )
556
551
agi = AllGatherBaseInfo (input_size = input .size (), codecs = codecs )
557
- # pyre-fixme[16]
558
552
AllGatherBase_Req .apply (group , myreq , agi , input )
559
553
return myreq
560
554
@@ -606,7 +600,6 @@ def reduce_scatter_v_pooled(
606
600
total_input_size = input_size ,
607
601
codecs = codecs ,
608
602
)
609
- # pyre-fixme[16]: `ReduceScatterV_Req` has no attribute `apply`.
610
603
ReduceScatterV_Req .apply (group , myreq , rsvi , input )
611
604
return myreq
612
605
0 commit comments