Skip to content

Commit b113eda

Browse files
committed
fix version
1 parent ed9b82e commit b113eda

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

torchao/testing/utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -320,7 +320,7 @@ def forward(self, x: torch.Tensor) -> torch.Tensor:
320320

321321
y_d = dn_dist(up_dist(input_dtensor))
322322

323-
if TORCH_VERSION_AT_LEAST_2_5:
323+
if not TORCH_VERSION_AT_LEAST_2_5:
324324
# Need torch 2.5 to support compiled tensor parallelism
325325
return
326326

0 commit comments

Comments
 (0)