From 3367b0f4310672903a7c2575d0c8c6e337f7188a Mon Sep 17 00:00:00 2001 From: "Alex Y. Chan" Date: Mon, 5 Oct 2026 14:20:00 +0100 Subject: [PATCH] Fix AArch64 NCCL EP linking with BOLT enabled Signed-off-by: Alex Y. Chan --- setup.py | 2 +- transformer_engine/common/CMakeLists.txt | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/setup.py b/setup.py index 8d0a87aaca3..2dabf37b322 100644 --- a/setup.py +++ b/setup.py @@ -285,7 +285,7 @@ def build_nccl_ep_submodule() -> str: if bolt_linker_flags: nvcc_linker_flags.extend(["-Xlinker=--emit-relocs", "-Xlinker=-z", "-Xlinker=now"]) if "-mno-fix-cortex-a53-843419" in bolt_linker_flags: - nvcc_linker_flags.append("-Xlinker=--no-fix-cortex-a53-843419") + nvcc_linker_flags.append("-Xcompiler=-mno-fix-cortex-a53-843419") def append_env_flags(name: str, flags: List[str]) -> None: if flags: diff --git a/transformer_engine/common/CMakeLists.txt b/transformer_engine/common/CMakeLists.txt index 529a5548a5a..843ec412c3d 100644 --- a/transformer_engine/common/CMakeLists.txt +++ b/transformer_engine/common/CMakeLists.txt @@ -447,7 +447,7 @@ if(NVTE_TARGET_IS_ARM64) target_link_options( transformer_engine PRIVATE - "LINKER:--no-fix-cortex-a53-843419" + "-mno-fix-cortex-a53-843419" ) endif()