Skip to content

Commit 5ca2a14

Browse files
generatedunixname89002005307016facebook-github-bot
authored andcommitted
suppress errors in torchrec (#887)
Summary: Pull Request resolved: #887 Differential Revision: D41897906 fbshipit-source-id: 8dcc31c96da6eda30ba8c7ab000eda75e72552c7
1 parent b5990fd commit 5ca2a14

File tree

1 file changed

+0
-7
lines changed

1 file changed

+0
-7
lines changed

torchrec/distributed/comm_ops.py

Lines changed: 0 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -320,7 +320,6 @@ def alltoall_pooled(
320320
cumsum_dim_sum_per_rank_tensor=cumsum_dim_sum_per_rank_tensor,
321321
codecs=codecs,
322322
)
323-
# pyre-fixme[16]: `All2All_Pooled_Req` has no attribute `apply`.
324323
All2All_Pooled_Req.apply(group, myreq, a2ai, a2a_pooled_embs_tensor)
325324
return myreq
326325

@@ -386,7 +385,6 @@ def alltoall_sequence(
386385
)
387386
# sequence of embeddings, bags are definitely non-uniform
388387

389-
# pyre-fixme[16]: `All2All_Seq_Req` has no attribute `apply`.
390388
All2All_Seq_Req.apply(group, myreq, a2ai, a2a_sequence_embs_tensor)
391389
return myreq
392390

@@ -448,7 +446,6 @@ def alltoallv(
448446
codecs=codecs,
449447
)
450448

451-
# pyre-fixme[16]: `All2Allv_Req` has no attribute `apply`.
452449
All2Allv_Req.apply(group, myreq, a2ai, inputs)
453450

454451
return myreq
@@ -486,7 +483,6 @@ def reduce_scatter_pooled(
486483
rsi = ReduceScatterInfo(
487484
input_sizes=[tensor.size() for tensor in inputs], codecs=codecs
488485
)
489-
# pyre-fixme[16]
490486
ReduceScatter_Req.apply(group, myreq, rsi, *inputs)
491487
return myreq
492488

@@ -520,7 +516,6 @@ def reduce_scatter_base_pooled(
520516

521517
myreq = Request(group, device=inputs.device)
522518
rsi = ReduceScatterBaseInfo(input_sizes=inputs.size(), codecs=codecs)
523-
# pyre-fixme[16]
524519
ReduceScatterBase_Req.apply(group, myreq, rsi, inputs)
525520
return myreq
526521

@@ -554,7 +549,6 @@ def all_gather_base_pooled(
554549

555550
myreq = Request(group, device=input.device)
556551
agi = AllGatherBaseInfo(input_size=input.size(), codecs=codecs)
557-
# pyre-fixme[16]
558552
AllGatherBase_Req.apply(group, myreq, agi, input)
559553
return myreq
560554

@@ -606,7 +600,6 @@ def reduce_scatter_v_pooled(
606600
total_input_size=input_size,
607601
codecs=codecs,
608602
)
609-
# pyre-fixme[16]: `ReduceScatterV_Req` has no attribute `apply`.
610603
ReduceScatterV_Req.apply(group, myreq, rsvi, input)
611604
return myreq
612605

0 commit comments

Comments
 (0)