Benchmarks: Fix Bug - Fix typo in gemm-flops benchmark. (#109)

1e96c27e · guoshzhao · GitHub · 8ffaddfa · 1e96c27e
Unverified Commit 1e96c27e authored Jun 30, 2021 by guoshzhao Committed by GitHub Jun 30, 2021
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

superbench/benchmarks/micro_benchmarks/gemm_flops_performance.py ...nch/benchmarks/micro_benchmarks/gemm_flops_performance.py +1 -1

No files found.
--- a/superbench/benchmarks/micro_benchmarks/gemm_flops_performance.py
+++ b/superbench/benchmarks/micro_benchmarks/gemm_flops_performance.py
@@ -111,7 +111,7 @@ class GemmFlopsCuda(MicroBenchmarkWithInvoke):
        else:
            self._args.precision = [p.upper() for p in self._args.precision]
            for p in self._args.precision:
-                if p not in list(self.__kernel_map.keys()):
+                if p not in self.__kernel_map[capability]:
                    self._result.set_return_code(ReturnCode.INVALID_ARGUMENT)
                    logger.warning(
                        'Unsupported precision - benchmark: {}, precision: {}, expected: {}.'.format(