Skip to content

Commit a5958e9

Browse files
committed
[Test][GPU] Skip float32 tensor core test when not supported
Only run tensor core FP32 tests when supported.
1 parent 7c0f701 commit a5958e9

File tree

1 file changed

+7
-1
lines changed

1 file changed

+7
-1
lines changed

max/kernels/test/gpu/layout/test_matmul.mojo

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -12,6 +12,7 @@
1212
# ===----------------------------------------------------------------------=== #
1313

1414
from sys import has_nvidia_gpu_accelerator
15+
from sys.info import _has_gpu_fp32_tensor_cores
1516

1617
from benchmark import Bench
1718
from buffer.dimlist import DimList
@@ -218,6 +219,11 @@ def main():
218219
test.run_test[k4](m)
219220
test.run_test[k5](m)
220221
test.run_test[k6](m)
221-
test_tc.run_test[k_tc](m)
222+
223+
@parameter
224+
if _has_gpu_fp32_tensor_cores():
225+
test_tc.run_test[k_tc](m)
226+
else:
227+
print("Skipping float32 tensor core test on GPU (not supported)")
222228

223229
m.dump_report()

0 commit comments

Comments
 (0)