[MI35x] fix gfx950 build error

hsa/gfx950/i8gemm/codegen.py

@@ -0,0 +1,75 @@
+# SPDX-License-Identifier: MIT
+# Copyright (C) 2018-2025, Advanced Micro Devices, Inc. All rights reserved.
+
+import os
+import argparse
+import glob
+import pandas as pd
+
+this_dir = os.path.dirname(os.path.abspath(__file__))
+hsa_dir = os.path.dirname(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
+
+template = """// SPDX-License-Identifier: MIT
+// Copyright (c) 2025, Advanced Micro Devices, Inc. All rights reserved.
+#pragma once
+#include <unordered_map>
+
+#define ADD_CFG(tile_m, tile_n, splitK, bpreshuffle, path, name, co)         \\
+    {                                                                        \\
+        name, { name, path co, tile_m, tile_n, splitK, bpreshuffle}          \\
+    }
+
+struct I8GemmConfig
+{
+    std::string name;
+    std::string co_name;
+    int tile_M;
+    int tile_N;
+    int splitK;
+    int bpreshuffle;
+};
+
+using CFG = std::unordered_map<std::string, I8GemmConfig>;
+
+"""
+if __name__ == "__main__":
+    parser = argparse.ArgumentParser(
+        prog="generate",
+        description="gen API for CK fmha kernel",
+    )
+    parser.add_argument(
+        "-o",
+        "--output_dir",
+        default="aiter/jit/build",
+        required=False,
+        help="write all the blobs into a directory",
+    )
+    args = parser.parse_args()
+
+    cfgs = []
+    for el in glob.glob(f"{this_dir}/*.csv"):
+        df = pd.read_csv(el)
+        cfg = [
+            f'ADD_CFG({tileM:>4}, {tileN:>4}, {splitK:>4}, {bpreshuffle:>4}, "i8gemm/", "{name}", "{co}"),'
+            for tileM, tileN, splitK, bpreshuffle, name, co in df.values
+        ]
+        filename = os.path.basename(el)
+        cfgname = filename.split(".")[0]
+        cfg_txt = "\n            ".join(cfg) + "\n"
+
+        txt = f"""static CFG cfg_{cfgname} = {{
+            {cfg_txt}}};"""
+        cfgs.append(txt)
+    if not cfgs:
+        for el in glob.glob(f"{hsa_dir}/gfx942/{os.path.basename(this_dir)}/*.csv"):
+            filename = os.path.basename(el)
+            cfgname = filename.split(".")[0]
+            cfg_txt = "\n"
+            cfgname = "i8gemm_bf16_perTokenI8"
+            txt = f"""static CFG cfg_{cfgname} = {{
+                {cfg_txt}}};"""
+            cfgs.append(txt)
+
+    txt_all = template + "\n".join(cfgs)
+    with open(f"{args.output_dir}/asm_i8gemm_configs.hpp", "w") as f:
+        f.write(txt_all)