bugfix for 1328

Author: zhiqchen-amd

3rdparty/aiter/aiter.patch

@@ -1,25 +1,14 @@
 --- aiter/jit/optCompilerConfig.json
 +++ aiter/jit/optCompilerConfig.json
-@@ -619,6 +619,7 @@
-         "verbose": "False",
+@@ -699,7 +699,7 @@
          "hip_clang_path": "os.environ.get('MHA_HIP_CLANG_PATH')",
          "blob_gen_cmd": [
-+            "f'{get_asm_dir()}/fmha_v3_fwd/codegen.py --output_dir {{}}'",
              "f'{CK_DIR}/example/ck_tile/01_fmha/generate.py -d fwd --receipt 600 --output_dir {{}}'",
-             "f'{AITER_CSRC_DIR}/cpp_itfs/mha_fwd_generate.py --receipt 3 --output_dir {{}}'"
+-            "f'{AITER_CSRC_DIR}/cpp_itfs/mha_fwd_generate.py --receipt 3 --output_dir {{}}'"
++            "f'{AITER_CSRC_DIR}/cpp_itfs/mha_fwd_generate.py --receipt 2 --output_dir {{}}'"
          ]
-
---- csrc/cpp_itfs/mha_fwd_generate.py
-+++ csrc/cpp_itfs/mha_fwd_generate.py
-@@ -150,7 +150,7 @@ COMBINED_API = """t = fmha_fwd_v3(traits, args, stream_config);
- API_MAP = {
-     1: FMHA_FWD_API.format(F_inner_dispatch=V3_API),
-     2: FMHA_FWD_API.format(F_inner_dispatch=V2_API),
--    3: FMHA_FWD_API.format(F_inner_dispatch=V2_API) + FMHA_FWD_SPLITKV_API,
-+    3: FMHA_FWD_API.format(F_inner_dispatch=COMBINED_API),
-     4: FMHA_BATCH_PREFILL_API,
-     5: FMHA_FWD_API.format(F_inner_dispatch=COMBINED_API)
-     + FMHA_FWD_SPLITKV_API
+     },
+     "module_mha_varlen_fwd": {
 
 --- csrc/py_itfs_cu/asm_pa.cu
 +++ csrc/py_itfs_cu/asm_pa.cu
@@ -89,6 +78,7 @@
  from . import mla
 +from . import paged_attn
 
+
 --- aiter/ops/gemm_op_a8w8.py
 +++ aiter/ops/gemm_op_a8w8.py
 @@ -425,9 +425,11 @@

open_source/bazel/arch_select.bzl

@@ -65,7 +65,7 @@ def subscribe_deps():
 def whl_deps():
     return select({
         "@//:using_cuda12": ["torch==2.6.0+cu126"],
-        "@//:using_rocm": ["pyrsmi", "amdsmi@https://sinian-metrics-platform.oss-cn-hangzhou.aliyuncs.com/kis%2FAMD%2Famd_smi%2Fali%2Famd_smi.tar", "aiter@https://sinian-metrics-platform.oss-cn-hangzhou.aliyuncs.com/kis/AMD/aiter/0.1.6/aiter-0.1.6%2Bgit.329d07ba.date.202511061625-py3-none-any.whl"],
+        "@//:using_rocm": ["pyrsmi", "amdsmi@https://sinian-metrics-platform.oss-cn-hangzhou.aliyuncs.com/kis%2FAMD%2Famd_smi%2Fali%2Famd_smi.tar", "aiter@https://sinian-metrics-platform.oss-cn-hangzhou.aliyuncs.com/kis/AMD/RTP/aiter-0.1.6%2Bgit.329d07ba.date.202511061023-py3-none-any.whl"],
         "//conditions:default": ["torch==2.1.2"],
     })
 

open_source/deps/http.bzl

@@ -60,9 +60,9 @@ def http_deps():
 
     http_archive(
         name = "aiter",
-        sha256 = "cf1ac18a72e08f38133cf8891a1484d694b482925f1196dda398fd10c19586f2",
+        sha256 = "4bd5179c70e5a2c95892db53a83bb50043fa79e5739eb54661ca6d451938963b",
         urls = [
-	        "https://sinian-metrics-platform.oss-cn-hangzhou.aliyuncs.com/kis/AMD/aiter/0.1.6/aiter-0.1.6%2Bgit.329d07ba.date.202511061625-py3-none-any.whl",
+	        "https://sinian-metrics-platform.oss-cn-hangzhou.aliyuncs.com/kis/AMD/RTP/aiter-0.1.6%2Bgit.329d07ba.date.202511061023-py3-none-any.whl",
 	    ],
         type = "zip",
         build_file = clean_dep("//:BUILD.aiter"),

open_source/deps/requirements_lock_rocm.txt

@@ -114,8 +114,8 @@ aiosignal==1.3.1 \
     --hash=sha256:54cd96e15e1649b75d6c87526a6ff0b6c1b0dd3459f43d9ca11d48c339b68cfc \
     --hash=sha256:f8376fb07dd1e86a584e4fcdec80b36b7f81aac666ebc724e2c090300dd83b17
     # via aiohttp
-aiter @ https://sinian-metrics-platform.oss-cn-hangzhou.aliyuncs.com/kis/AMD/aiter/0.1.6/aiter-0.1.6%2Bgit.329d07ba.date.202511061625-py3-none-any.whl \
-    --hash=sha256:cf1ac18a72e08f38133cf8891a1484d694b482925f1196dda398fd10c19586f2
+aiter @ https://sinian-metrics-platform.oss-cn-hangzhou.aliyuncs.com/kis/AMD/RTP/aiter-0.1.6%2Bgit.329d07ba.date.202511061023-py3-none-any.whl \
+    --hash=sha256:4bd5179c70e5a2c95892db53a83bb50043fa79e5739eb54661ca6d451938963b
     # via -r open_source/deps/requirements_rocm.txt
 aliyun-python-sdk-core==2.15.2 \
     --hash=sha256:54f66a53e193c61c5e16ea4505a0cab43543f8ad2ef22833f69c4d5e5151c17d

open_source/deps/requirements_rocm.txt

@@ -4,5 +4,5 @@ https://repo.radeon.com/rocm/manylinux/rocm-rel-6.4.1/torch-2.4.1%2Brocm6.4.1.gi
 https://repo.radeon.com/rocm/manylinux/rocm-rel-6.4.1/torchvision-0.19.0%2Brocm6.4.1.git4d41ad71-cp310-cp310-linux_x86_64.whl
 pyrsmi
 pyyaml
-https://sinian-metrics-platform.oss-cn-hangzhou.aliyuncs.com/kis/AMD/aiter/0.1.6/aiter-0.1.6%2Bgit.329d07ba.date.202511061625-py3-none-any.whl
+https://sinian-metrics-platform.oss-cn-hangzhou.aliyuncs.com/kis/AMD/RTP/aiter-0.1.6%2Bgit.329d07ba.date.202511061023-py3-none-any.whl
 https://sinian-metrics-platform.oss-cn-hangzhou.aliyuncs.com/kis%2FAMD%2Famd_smi%2Fali%2Famd_smi.tar