Changes for gcc-toolset-12 (#125)

open-ce · Feb 28, 2024 · b04a4bc · b04a4bc
1 parent 7d727ff
commit b04a4bc
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 6 deletions.
diff --git a/pytorch-2.1.2/recipe/build.sh b/pytorch-2.1.2/recipe/build.sh
@@ -18,12 +18,12 @@ set -ex
 
 if [[ $ppc_arch == "p10" ]]
 then
-  if [[ -z "${GCC_11_HOME}" ]];
+  if [[ -z "${GCC_HOME}" ]];
   then
-    echo "Please set GCC_11_HOME to the install path of gcc-toolset-11"
+    echo "Please set GCC_HOME to the install path of gcc-toolset-12"
     exit 1
   else
-    export PATH=${GCC_11_HOME}/bin/:$PATH
+    export PATH=${GCC_HOME}/bin/:$PATH
     rm ${PREFIX}/lib/libstdc++.so*
     rm ${BUILD_PREFIX}/lib/libstdc++.so*
   fi

diff --git a/pytorch-2.1.2/recipe/meta.yaml b/pytorch-2.1.2/recipe/meta.yaml
@@ -61,7 +61,6 @@ requirements:
     - typing_extensions {{ typing_extensions }}
     - click {{ click }}
     # https://github.com/AnacondaRecipes/click-feedstock/issues/4
-    - importlib-metadata                     # [py<38]
     - networkx {{ networkx }}
     - scipy {{ scipy }}
     - filelock
@@ -85,7 +84,7 @@ requirements:
     - tensorboard {{ tensorboard }}
 
 build:
-  number: 4
+  number: 5
 {% if build_type == 'cpu' %}
   string: {{ build_type }}_py{{ python | replace(".", "") }}_pb{{ protobuf | replace(".*", "")}}_{{ PKG_BUILDNUM }}
 {% else %}
@@ -94,7 +93,7 @@ build:
   detect_binary_files_with_prefix: False
   script_env:
     - CUDA_HOME                                  #[build_type == 'cuda']
-    - GCC_11_HOME                                #[ppc_arch == 'p10']
+    - GCC_HOME                                #[ppc_arch == 'p10']
 
 test:
   imports: