[Tests] Check int8+int32 testcases in test_estimate_peak_flops_cpu (#…

…16019) Pytest param fixture for dtypes was not used resulting in these two test cases not beeing checked.
apache · Oct 31, 2023 · 5b561a6 · 5b561a6
1 parent c4c0a49
commit 5b561a6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/python/unittest/test_roofline.py b/tests/python/unittest/test_roofline.py
@@ -42,7 +42,7 @@ def test_estimate_peak_flops_cpu(dtype):
     target = tvm.target.Target("llvm -mattr=+fma,+avx2")
     dev = remote.device(str(target))
     # This test uses vectorized instructions so we need a target that supports them
-    flops = tvm.utils.roofline.x86.estimate_peak_fma_vector_flops(target, dev, remote, "float32")
+    flops = tvm.utils.roofline.x86.estimate_peak_fma_vector_flops(target, dev, remote, dtype)
     # Assume we can achieve 1 GFLOP/s per thread, which is 1 FLOP per cycle on a 1GHz cpu.
     assert (
         flops > 10**9 and flops < 10**14