From 5e91d8444f2880cdbe83a4d5f09fc477a49e1cc4 Mon Sep 17 00:00:00 2001
From: mtgu0705 <mtgu@amd.com>
Date: Mon, 30 Dec 2024 19:47:30 +0800
Subject: [PATCH] Change constant to  variable

---
 .../gpu/block/blockwise_gemm_pipeline_xdlops_v3_ab_scale.hpp  | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/include/ck/tensor_operation/gpu/block/blockwise_gemm_pipeline_xdlops_v3_ab_scale.hpp b/include/ck/tensor_operation/gpu/block/blockwise_gemm_pipeline_xdlops_v3_ab_scale.hpp
index 73fb67eda0..15f70698f2 100644
--- a/include/ck/tensor_operation/gpu/block/blockwise_gemm_pipeline_xdlops_v3_ab_scale.hpp
+++ b/include/ck/tensor_operation/gpu/block/blockwise_gemm_pipeline_xdlops_v3_ab_scale.hpp
@@ -339,7 +339,7 @@ struct BlockwiseGemmXdlops_pipeline_v3_ab_scale<BlockGemmPipelineScheduler::Intr
         b_blockwise_copy.MoveSrcSliceWindow(b_grid_desc, b_block_copy_step);
 
         static_for<0, MRepeat, 1>{}([&](auto m0) {
-            static_for<0, Number<16>{}, 1>{}([&](auto i0) {
+            static_for<0, Number<xdlops_gemm.GetRegSizePerXdlops()>{}, 1>{}([&](auto i0) {
                 a_scale_thread_copy.Run(a_scale_grid_desc,
                                         a_scale_grid_buf,
                                         a_scale_thread_desc,
@@ -479,7 +479,7 @@ struct BlockwiseGemmXdlops_pipeline_v3_ab_scale<BlockGemmPipelineScheduler::Intr
                     });
                 });
                 static_for<0, MRepeat, 1>{}([&](auto m0) {
-                    static_for<0, Number<16>{}, 1>{}([&](auto i0) {
+                    static_for<0, Number<xdlops_gemm.GetRegSizePerXdlops()>{}, 1>{}([&](auto i0) {
                         a_scale_thread_copy.Run(a_scale_grid_desc,
                                                 a_scale_grid_buf,
                                                 a_scale_thread_desc,