Skip to content

Commit cdf48f0

Browse files
Fix broken test_affine_quantized_tensor_parallel test after DeviceMesh (#3151)
fix broken test_affine_quantized_tensor_parallel test after DeviceMesh update
1 parent fa17427 commit cdf48f0

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

test/dtypes/test_affine_quantized_tensor_parallel.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -115,7 +115,7 @@ def forward(self, x: torch.Tensor) -> torch.Tensor:
115115
dn_quant(up_quant(example_input))
116116

117117
mesh = self.build_device_mesh()
118-
mesh.device_type = "cuda"
118+
mesh._device_type = "cuda"
119119

120120
# Shard the models
121121
up_dist = self.colwise_shard(up_quant, mesh)

0 commit comments

Comments
 (0)