use libnvjitlink package for libnvjitlink.so (#128)

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

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,14 +79,15 @@ requirements:
     - cudatoolkit {{ cudatoolkit }}
     - cudnn {{ cudnn }}
     - nccl {{ nccl }}
+    - libnvjitlink
    #- tensorrt {{ tensorrt }}                    #[py<38]
 {% endif %}
 
   run_constrained:
     - tensorboard {{ tensorboard }}
 
 build:
-  number: 5
+  number: 6
 {% if build_type == 'cpu' %}
   string: {{ build_type }}_py{{ python | replace(".", "") }}_pb{{ protobuf | replace(".*", "")}}_{{ PKG_BUILDNUM }}
 {% else %}