From bcfd271764e73b690758105f686b2e0c2875defd Mon Sep 17 00:00:00 2001
From: Deepali <70963368+cdeepali@users.noreply.github.com>
Date: Fri, 12 Apr 2024 10:57:13 +0530
Subject: [PATCH] use libnvjitlink package for libnvjitlink.so (#128)

---
 pytorch-2.1.2/recipe/build.sh  | 5 -----
 pytorch-2.1.2/recipe/meta.yaml | 6 ++++--
 2 files changed, 4 insertions(+), 7 deletions(-)

diff --git a/pytorch-2.1.2/recipe/build.sh b/pytorch-2.1.2/recipe/build.sh
index 77857fb..91d1097 100644
--- a/pytorch-2.1.2/recipe/build.sh
+++ b/pytorch-2.1.2/recipe/build.sh
@@ -127,11 +127,6 @@ then
   find /usr/include -name cublas*.h -exec ln -s "{}" "$CONDA_PREFIX/include/" ';'
   export CXXFLAGS="${CXXFLAGS} -I${PREFIX}/include -I${CUDA_HOME}/include -I${CONDA_PREFIX}/include"
 
-  # Create symlinks for livnvJitLink lib for cuda 12.2
-  if [[ $cudatoolkit == '12.2' ]]; then
-    find ${CUDA_HOME}/targets/${ARCH}-linux/lib/libnvJitLink.* -exec ln -s "{}" "$PREFIX/lib/" ';'
-  fi
-
   # Temporarily disable TensorRT
   #if [[ $PY_VER < 3.9 ]]
   #then
diff --git a/pytorch-2.1.2/recipe/meta.yaml b/pytorch-2.1.2/recipe/meta.yaml
index 55d0500..4902f23 100644
--- a/pytorch-2.1.2/recipe/meta.yaml
+++ b/pytorch-2.1.2/recipe/meta.yaml
@@ -12,7 +12,7 @@ source:
 #    - 0104-onnx-tensorrt-TRT7.patch                  # [py<39]
     - 0106-disable-SHM-and-CMA-for-tensorpipe.patch
 #    - 0107-fix-clock-gettime-undefind-error-for-sleef.patch
-#    - 0108-skip-half-test.patch         # [x86_64]
+    - 0108-skip-half-test.patch          # [ppc64le]
 
     # 03xx - patch temporary to fix a problem that when fixed upstream can be removed
     - 0309-fallback-to-cpu_kernel-with-VSX.patch     # [ppc64le]
@@ -49,6 +49,7 @@ requirements:
     - cudnn {{ cudnn }}
     - nccl {{ nccl }}
     - magma {{ magma }}
+    - libnvjitlink
    #- tensorrt {{ tensorrt }}                #[py<38]
 {% endif %}
 
@@ -78,6 +79,7 @@ requirements:
     - cudatoolkit {{ cudatoolkit }}
     - cudnn {{ cudnn }}
     - nccl {{ nccl }}
+    - libnvjitlink
    #- tensorrt {{ tensorrt }}                    #[py<38]
 {% endif %}
 
@@ -85,7 +87,7 @@ requirements:
     - tensorboard {{ tensorboard }}
 
 build:
-  number: 5
+  number: 6
 {% if build_type == 'cpu' %}
   string: {{ build_type }}_py{{ python | replace(".", "") }}_pb{{ protobuf | replace(".*", "")}}_{{ PKG_BUILDNUM }}
 {% else %}