skip smooth quant test (torch dynamo)

petrex · petrex · commit bbf74f987d0f · 2025-01-24T09:37:28.000-08:00
diff --git a/test/prototype/test_smoothquant.py b/test/prototype/test_smoothquant.py
@@ -20,6 +20,9 @@
     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,9 @@`
`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)`
	`25`	`+`
`23`	`26`
`24`	`27`	`class ToyLinearModel(torch.nn.Module):`
`25`	`28`	`def __init__(self, m=512, n=256, k=128):`