wip

pytorch · Feb 17, 2024 · 3190b89 · 3190b89
1 parent eeb3a43
commit 3190b89
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/.github/scripts/utils_build.bash b/.github/scripts/utils_build.bash
@@ -104,8 +104,8 @@ __conda_install_clang () {
 
   echo "[INSTALL] Installing Clang through Conda ..."
   # shellcheck disable=SC2086
-  # (exec_with_retries 3 conda install ${env_prefix} -c conda-forge -y clangxx=${llvm_version} libcxx=${llvm_version} llvm-openmp=${llvm_version} compiler-rt=${llvm_version} ) || return 1
-  (exec_with_retries 3 conda install ${env_prefix} -c conda-forge -y clangxx=${llvm_version} ) || return 1
+  # (exec_with_retries 3 conda install ${env_prefix} -c conda-forge -y clangxx=${llvm_version} libcxx=${llvm_version} llvm-openmp=${llvm_version} compiler-rt=${llvm_version}) || return 1
+  (exec_with_retries 3 conda install ${env_prefix} -c conda-forge -y clangxx=${llvm_version}) || return 1
 
   # The compilers are visible in the PATH as `clang` and `clang++`, so symlinks
   # will need to be created

diff --git a/fbgemm_gpu/setup.py b/fbgemm_gpu/setup.py
@@ -186,8 +186,8 @@ def _get_cxx11_abi():
             [
                 f"-DCMAKE_C_COMPILER={path}/bin/cc",
                 f"-DCMAKE_CXX_COMPILER={path}/bin/c++",
-                f"-DCMAKE_C_FLAGS='-fopenmp -stdlib=libstdc++ -I{path}/include'",
-                f"-DCMAKE_CXX_FLAGS='-fopenmp -stdlib=libstdc++ -I{path}/include'",
+                f"-DCMAKE_C_FLAGS='-fopenmp=libgomp -stdlib=libstdc++ -I{path}/include'",
+                f"-DCMAKE_CXX_FLAGS='-fopenmp=libgomp -stdlib=libstdc++ -I{path}/include'",
                 # f"-DOpenMP_C_LIB_NAMES={name}",
                 # f"-DOpenMP_C_FLAGS=-fopenmp={name}",
                 # f"-DCMAKE_C_FLAGS='-fopenmp={name} -I{path}/include'",