wip

pytorch · Nov 15, 2024 · d711159 · d711159
1 parent 634dde0
commit d711159
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git 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
@@ -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 ...")