From d711159b3fd35169b56fffd4610b7c1f547f789a Mon Sep 17 00:00:00 2001 From: Benson Ma Date: Thu, 14 Nov 2024 15:08:25 -0800 Subject: [PATCH] wip --- .github/scripts/fbgemm_gpu_build.bash | 4 ++-- fbgemm_gpu/setup.py | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/.github/scripts/fbgemm_gpu_build.bash b/.github/scripts/fbgemm_gpu_build.bash index 1a7ca88264..254cd60fff 100644 --- a/.github/scripts/fbgemm_gpu_build.bash +++ b/.github/scripts/fbgemm_gpu_build.bash @@ -199,9 +199,9 @@ __configure_fbgemm_gpu_build_cuda () { # https://github.com/NVIDIA/nvbench/discussions/129 # https://github.com/vllm-project/vllm/blob/main/CMakeLists.txt#L187 # https://github.com/NVIDIA/cutlass/blob/main/include/cutlass/gemm/kernel/sm90_gemm_tma_warpspecialized.hpp#L224 - local arch_list="7.0;8.0;9.0;9.0a" + local arch_list="8.0;9.0;9.0a" else - local arch_list="7.0;8.0" + local arch_list="8.0" fi fi echo "[BUILD] Setting the following CUDA targets: ${arch_list}" diff --git a/fbgemm_gpu/setup.py b/fbgemm_gpu/setup.py index 53bd0377fa..1ace81b393 100644 --- a/fbgemm_gpu/setup.py +++ b/fbgemm_gpu/setup.py @@ -247,7 +247,7 @@ def _get_cxx11_abi(): _get_cxx11_abi(), ] - cxx_flags = ["-mcmodel=large", "-fuse-linker-plugin", "-flto"] + cxx_flags = ["-mcmodel=large", "-flto"] if self.args.verbose: print("[SETUP.PY] Building in VERBOSE mode ...")