Skip to content

Commit 19ac29e

Browse files
add a fp64 skip case (#2123) (#2128)
Co-authored-by: Weishi.Deng <[email protected]>
1 parent 456459a commit 19ac29e

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

tests/gpu/examples/test_fake_quantize.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -53,6 +53,7 @@ def test_fake_quantize_learnable_per_channel_affine(self, dtype=torch.float):
5353
channel_zero_point_xpu, 1, quant_min, quant_max, grad_factor)
5454
self.assertEqual(dst_cpu, dst_xpu.cpu())
5555

56+
@pytest.mark.skipif(not torch.xpu.utils.has_fp64_dtype(), reason="fp64 not support by this device")
5657
def test_fake_quantize_learnable_per_channel_affine_backward(self, dtype=torch.float):
5758
x_cpu = torch.randn([1, 3, 2, 2], requires_grad=True)
5859
x_xpu = x_cpu.clone().detach().to("xpu")

0 commit comments

Comments
 (0)