Skip to content

Commit 71c84c8

Browse files
committed
fix
1 parent 8b12114 commit 71c84c8

File tree

1 file changed

+0
-4
lines changed

1 file changed

+0
-4
lines changed

tests/gemm/test_cute_dsl_gemm_allreduce_two_shot.py

Lines changed: 0 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -43,10 +43,6 @@ def create_mc_tensor(torch_tensor_cpu, dtype, leading_dim, is_dynamic_layout=Tru
4343
torch_symm_tensor.copy_(torch_tensor_cpu)
4444
symm = symm_mem.rendezvous(torch_symm_tensor, group=dist.group.WORLD.group_name)
4545
mc_ptr = symm.multicast_ptr
46-
47-
if not mc_ptr:
48-
raise ValueError("Multicast support is not available")
49-
5046
# create MC tensor memref
5147
cute_tensor_mc = from_dlpack(
5248
cutlass_torch.as_tensor(mc_ptr, torch_tensor_cpu.shape, torch_tensor_cpu.dtype),

0 commit comments

Comments
 (0)