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()