From 14ed75f7d396ef6d247f451749bf139d8edc68cb Mon Sep 17 00:00:00 2001 From: ming1753 <61511741+ming1753@users.noreply.github.com> Date: Tue, 5 Aug 2025 15:25:28 +0800 Subject: [PATCH] [Test] scaled_gemm_f8_i4_f16 skip test while sm != 89 (#3210) --- test/operators/test_scaled_gemm_f8_i4_f16.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/test/operators/test_scaled_gemm_f8_i4_f16.py b/test/operators/test_scaled_gemm_f8_i4_f16.py index a154d1df8..dd80b3a6e 100644 --- a/test/operators/test_scaled_gemm_f8_i4_f16.py +++ b/test/operators/test_scaled_gemm_f8_i4_f16.py @@ -33,6 +33,10 @@ class Test(unittest.TestCase): paddle.seed(2024) print(paddle.device.cuda.get_device_properties()) print(paddle.__git_commit__) + prop = paddle.device.cuda.get_device_properties() + cc = prop.major * 10 + prop.minor + if cc != 89: + self.skipTest("scaled_gemm_f8_i4_f16 only support sm 89!") def quant_fp8_pertensor(self, tensor): """