Fix the issue with bias_size in compile_kernels

ROCm · Jul 22, 2024 · 1daec1f · 1daec1f
1 parent 85f7d93
commit 1daec1f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/scripts/amd/gemm/utils/file_generator.py b/scripts/amd/gemm/utils/file_generator.py
@@ -230,7 +230,7 @@ def main():
     numThreads = args.n
     rotating_buffer_size = args.rotating_tensor
     """
-    compile_kernels_call_str = f'compile_kernels({M}, {N}, {K}, rotating_buffer_size, {M}, numThreads)'
+    compile_kernels_call_str = f'compile_kernels({M}, {N}, {K}, rotating_buffer_size, {bias_size}, numThreads)'
 
     f_kernel.write(def_main_str)
     f_kernel.write(compile_kernels_call_str + "\n\n")