Waive test_quantization.py NVbug 5170222

Fridah-nv · web-flow · commit f6591d574a8e · 2025-07-31T13:37:52.000-07:00
Signed-off-by: Fridah-nv &lt;201670829+Fridah-nv@users.noreply.github.com&gt;
diff --git a/tests/unittest/_torch/auto_deploy/unit/singlegpu/transformations/library/test_quantization.py b/tests/unittest/_torch/auto_deploy/unit/singlegpu/transformations/library/test_quantization.py
@@ -56,7 +56,7 @@ def get_quant_config(self):
     ],
 )
 def test_quantization(quant_config, atol, rtol, num_p_og):
-    # pytest.skip("https://nvbugspro.nvidia.com/bug/5170222")
+    pytest.skip("https://nvbugspro.nvidia.com/bug/5170222")
     model = MLP(32, 64, 32).to(torch.float16).to("cuda")
     x = torch.randn(3, 32, dtype=torch.float16).to("cuda")
 

Original file line number	Diff line number	Diff line change
`@@ -56,7 +56,7 @@ def get_quant_config(self):`
`56`	`56`	`],`
`57`	`57`	`)`
`58`	`58`	`def test_quantization(quant_config, atol, rtol, num_p_og):`
`59`		`- # pytest.skip("https://nvbugspro.nvidia.com/bug/5170222")`
	`59`	`+ pytest.skip("https://nvbugspro.nvidia.com/bug/5170222")`
`60`	`60`	`model = MLP(32, 64, 32).to(torch.float16).to("cuda")`
`61`	`61`	`x = torch.randn(3, 32, dtype=torch.float16).to("cuda")`
`62`	`62`