We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 7c0f701 commit a5958e9Copy full SHA for a5958e9
max/kernels/test/gpu/layout/test_matmul.mojo
@@ -12,6 +12,7 @@
12
# ===----------------------------------------------------------------------=== #
13
14
from sys import has_nvidia_gpu_accelerator
15
+from sys.info import _has_gpu_fp32_tensor_cores
16
17
from benchmark import Bench
18
from buffer.dimlist import DimList
@@ -218,6 +219,11 @@ def main():
218
219
test.run_test[k4](m)
220
test.run_test[k5](m)
221
test.run_test[k6](m)
- 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)")
228
229
m.dump_report()
0 commit comments