[Issue]: TypeError: `get_gemm1_kernels_list()` and `get_gemm2_kernels_list()` missing 1 required positional argument: 'preshuffle' when running `hsa/gfx942//fmoe_2stages/tune.py`

[tjtanaa]

Problem Description

Commit a7f63e3 introduced preshuffle to the get_gemm1_kernels_list in

aiter/csrc/ck_gemm_moe_2stages_codegen/gemm_moe_ck2stages_common.py

Line 332 in 288c82f

preshuffle: bool,

  File "/app/debugds/update_fmoe_dsv3_ptpc_config/hsa/gfx942//fmoe_2stages/tune.py", line 1518, in gen_2stages_task                                                                                                 
    _, ck_stage1_kernels = get_gemm1_kernels_list(                                                                                                                                                                  
                           ^^^^^^^^^^^^^^^^^^^^^^^                                                                                                                                                                  
TypeError: get_gemm1_kernels_list() missing 1 required positional argument: 'preshuffle'         

and

Traceback (most recent call last):                                                                                                                                                                                  
  File "/app/debugds/update_fmoe_dsv3_ptpc_config/aiter/utility/base_tuner.py", line 386, in run                                                                                                                    
    all_results = self.tune(batch, self.tunedf, args)                                                                                                                                                               
                  ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^                                                                                                                                                               
  File "/app/debugds/update_fmoe_dsv3_ptpc_config/hsa/gfx942//fmoe_2stages/tune.py", line 1699, in tune                                                                                                             
    tasks_ck.extend(self.gen_2stages_task(info, blockMs))                                                                                                                                                           
                    ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^                                                                                                                                                            
  File "/app/debugds/update_fmoe_dsv3_ptpc_config/hsa/gfx942//fmoe_2stages/tune.py", line 1528, in gen_2stages_task                                                                                                 
    _, ck_stage2_kernels = get_gemm2_kernels_list(                                                                                                                                                                  
                           ^^^^^^^^^^^^^^^^^^^^^^^                                                                                                                                                                  
TypeError: get_gemm2_kernels_list() missing 1 required positional argument: 'preshuffle'

Operating System

Ubuntu 22.04.4 LTS (Jammy Jellyfish)

CPU

AMD EPYC 9654 96-Core Processor

GPU

AMD Instinct MI300X

ROCm Version

ROCm 7.0

ROCm Component

No response

Steps to Reproduce

No response

(Optional for Linux users) Output of /opt/rocm/bin/rocminfo --support

No response

Additional Information

No response