diff --git a/extra/mmapeak/mmapeak.py b/extra/mmapeak/mmapeak.py index 37b70c8c6e..05ee3010a7 100644 --- a/extra/mmapeak/mmapeak.py +++ b/extra/mmapeak/mmapeak.py @@ -48,7 +48,7 @@ if __name__=="__main__": COMPILER = HIPCompiler(DEV.arch) if DEV.arch in {'gfx1100', 'gfx1103', 'gfx1151'}: if DEV.arch == 'gfx1103': NUM_WORKGROUPS = 8 - if DEV.arch == 'gfx1151': NUM_WORKGROUPS = 40 + if DEV.arch == 'gfx1151': NUM_WORKGROUPS = 32 launchBenchmark("v_wmma_bf16_16x16x16_bf16", (7,8,15)) launchBenchmark("v_wmma_f16_16x16x16_f16", (7,8,15)) launchBenchmark("v_wmma_f32_16x16x16_bf16", (7,8,15))