From 964557610c6648863cbdd2aae96a7fd6a260ad69 Mon Sep 17 00:00:00 2001 From: Daniel Vega-Myhre Date: Fri, 10 Oct 2025 12:58:38 -0700 Subject: [PATCH] fix broken test_affine_quantized_tensor_parallel test after DeviceMesh update --- test/dtypes/test_affine_quantized_tensor_parallel.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/test/dtypes/test_affine_quantized_tensor_parallel.py b/test/dtypes/test_affine_quantized_tensor_parallel.py index 983f701849..5a6e89c25f 100644 --- a/test/dtypes/test_affine_quantized_tensor_parallel.py +++ b/test/dtypes/test_affine_quantized_tensor_parallel.py @@ -115,7 +115,7 @@ def forward(self, x: torch.Tensor) -> torch.Tensor: dn_quant(up_quant(example_input)) mesh = self.build_device_mesh() - mesh.device_type = "cuda" + mesh._device_type = "cuda" # Shard the models up_dist = self.colwise_shard(up_quant, mesh)