Skip to content

Commit 6b356b3

Browse files
committed
fix
1 parent 0f8289d commit 6b356b3

File tree

1 file changed

+0
-4
lines changed

1 file changed

+0
-4
lines changed

tests/unlisted/test_cute_dsl_gemm_allreduce_two_shot.py

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

0 commit comments

Comments
 (0)