skip smooth quant test (torch dynamo)

petrex · petrex · commit f0ec4b329c25 · 2025-01-24T09:35:09.000-08:00
diff --git a/test/prototype/test_smoothquant.py b/test/prototype/test_smoothquant.py
@@ -20,6 +20,8 @@
     TORCH_VERSION_AT_LEAST_2_5,
 )
 
+if torch.version.hip is not None:
+    pytest.skip("Skipping the test in ROCm", allow_module_level=True)
 
 class ToyLinearModel(torch.nn.Module):
     def __init__(self, m=512, n=256, k=128):

Original file line number	Diff line number	Diff line change
`@@ -20,6 +20,8 @@`
`20`	`20`	`TORCH_VERSION_AT_LEAST_2_5,`
`21`	`21`	`)`
`22`	`22`
	`23`	`+if torch.version.hip is not None:`
	`24`	`+ pytest.skip("Skipping the test in ROCm", allow_module_level=True)`
`23`	`25`
`24`	`26`	`class ToyLinearModel(torch.nn.Module):`
`25`	`27`	`def __init__(self, m=512, n=256, k=128):`