Skip to content

Commit 5b561a6

Browse files
authored
[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.
1 parent c4c0a49 commit 5b561a6

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

tests/python/unittest/test_roofline.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -42,7 +42,7 @@ def test_estimate_peak_flops_cpu(dtype):
4242
target = tvm.target.Target("llvm -mattr=+fma,+avx2")
4343
dev = remote.device(str(target))
4444
# This test uses vectorized instructions so we need a target that supports them
45-
flops = tvm.utils.roofline.x86.estimate_peak_fma_vector_flops(target, dev, remote, "float32")
45+
flops = tvm.utils.roofline.x86.estimate_peak_fma_vector_flops(target, dev, remote, dtype)
4646
# Assume we can achieve 1 GFLOP/s per thread, which is 1 FLOP per cycle on a 1GHz cpu.
4747
assert (
4848
flops > 10**9 and flops < 10**14

0 commit comments

Comments
 (0)