diff --git a/xllm/compiler/tilelang/targets/ascend/kernel_family_builder.py b/xllm/compiler/tilelang/targets/ascend/kernel_family_builder.py index ab5ac75b62..55426db5af 100644 --- a/xllm/compiler/tilelang/targets/ascend/kernel_family_builder.py +++ b/xllm/compiler/tilelang/targets/ascend/kernel_family_builder.py @@ -107,7 +107,7 @@ def _run_variant_worker(args: _VariantWorkerArgs) -> _VariantBuildResult: compile_cmd = [ args.bisheng_executable, *args.bisheng_compile_flags, - f"-Dg_tilingKey=g_tilingKey__{compile_spec.variant_key}", + f"-Dg_tilingKey=g_tilingKey__{compile_spec.kernel_name}_{compile_spec.variant_key}", *[f"-I{d}" for d in args.include_dirs], str(plan.generated_source), "-c",