Add libcusolver, libcusparse, and libnvjitlink recipes

recipes/libcusolver/bld.bat

@@ -0,0 +1,5 @@
+if not exist %PREFIX% mkdir %PREFIX%
+
+move lib\x64\* %LIBRARY_LIB%
+move bin\* %LIBRARY_BIN%
+move include\* %LIBRARY_INC%

recipes/libcusolver/build.sh

@@ -0,0 +1,33 @@
+#!/bin/bash
+
+# Install to conda style directories
+[[ -d lib64 ]] && mv lib64 lib
+mkdir -p ${PREFIX}/lib
+[[ -d pkg-config ]] && mv pkg-config ${PREFIX}/lib/pkgconfig
+[[ -d "$PREFIX/lib/pkgconfig" ]] && sed -E -i "s|cudaroot=.+|cudaroot=$PREFIX|g" $PREFIX/lib/pkgconfig/cusolver*.pc
+
+[[ ${target_platform} == "linux-64" ]] && targetsDir="targets/x86_64-linux"
+[[ ${target_platform} == "linux-ppc64le" ]] && targetsDir="targets/ppc64le-linux"
+[[ ${target_platform} == "linux-aarch64" ]] && targetsDir="targets/sbsa-linux"
+
+for i in `ls`; do
+    [[ $i == "build_env_setup.sh" ]] && continue
+    [[ $i == "conda_build.sh" ]] && continue
+    [[ $i == "metadata_conda_debug.yaml" ]] && continue
+    if [[ $i == "lib" ]] || [[ $i == "include" ]]; then
+        # Headers and libraries are installed to targetsDir
+        mkdir -p ${PREFIX}/${targetsDir}
+        mkdir -p ${PREFIX}/$i
+        cp -rv $i ${PREFIX}/${targetsDir}
+        if [[ $i == "lib" ]]; then
+            for j in "$i"/*.so*; do
+                # Shared libraries are symlinked in $PREFIX/lib
+                ln -s ${PREFIX}/${targetsDir}/$j ${PREFIX}/$j
+            done
+        fi
+    else
+        # Put all other files in targetsDir
+        mkdir -p ${PREFIX}/${targetsDir}/${PKG_NAME}
+        cp -rv $i ${PREFIX}/${targetsDir}/${PKG_NAME}
+    fi
+done

recipes/libcusolver/conda_build_config.yaml

@@ -0,0 +1,2 @@
+arm_variant_type:  # [aarch64]
+  - sbsa           # [aarch64]

recipes/libcusolver/meta.yaml

@@ -0,0 +1,185 @@
+{% set name = "libcusolver" %}
+{% set version = "11.4.2.57" %}
+{% set cuda_version = "12.0" %}
+{% set platform = "linux-x86_64" %}  # [linux64]
+{% set platform = "linux-ppc64le" %}  # [ppc64le]
+{% set platform = "linux-sbsa" %}  # [aarch64]
+{% set platform = "windows-x86_64" %}  # [win]
+{% set target_name = "x86_64-linux" %}  # [linux64]
+{% set target_name = "ppc64le-linux" %}  # [ppc64le]
+{% set target_name = "sbsa-linux" %}  # [aarch64]
+{% set target_name = "x64" %}  # [win]
+{% set extension = "tar.xz" %}  # [not win]
+{% set extension = "zip" %}  # [win]
+
+package:
+  name: {{ name|lower }}
+  version: {{ version }}
+
+source:
+  url: https://developer.download.nvidia.com/compute/cuda/redist/{{ name }}/{{ platform }}/{{ name }}-{{ platform }}-{{ version }}-archive.{{ extension }}
+  sha256: c9460db500d0c7d14debb2c8e0d88fc26a39197363f1b000ccf760866f636f12  # [linux64]
+  sha256: bf3c9a090652d011aafe2190c6f26fe108194cf8512d3c571e406054740168bb  # [ppc64le]
+  sha256: 0ca46b73520657f96f921711c05af7053bd398328d510aecddfff2521c000e91  # [aarch64]
+  sha256: c09836df1f43a3496f2b61ad8827af662ace298508869564a40c30d909ba550f  # [win]
+
+build:
+  number: 0
+  skip: true  # [osx]
+
+outputs:
+  - name: libcusolver
+    files:
+      - lib/libcusolver*.so.*                            # [linux]
+      - targets/{{ target_name }}/lib/libcusolver*.so.*  # [linux]
+      - Library\bin\cusolver*.dll                        # [win]
+    requirements:
+      build:
+        - {{ compiler("c") }}
+        - {{ compiler("cxx") }}
+        - arm-variant * {{ arm_variant_type }}  # [aarch64]
+        - sysroot_{{ target_platform }} 2.17    # [linux]
+      host:
+        - cuda-version {{ cuda_version }}
+        - libcublas
+        - libnvjitlink
+      run:
+        - {{ pin_compatible("cuda-version", max_pin="x.x") }}
+        - {{ pin_compatible("libcublas", max_pin="x.x") }}
+        - {{ pin_compatible("libnvjitlink", max_pin="x.x") }}
+      run_constrained:
+        - arm-variant * {{ arm_variant_type }}  # [aarch64]
+    test:
+      commands:
+        - test -L $PREFIX/lib/libcusolver.so.{{ version }}                  # [linux]
+        - test -L $PREFIX/lib/libcusolver.so.{{ version.split(".")[0] }}    # [linux]
+        - test -L $PREFIX/targets/{{ target_name }}/lib/libcusolver.so.{{ version }}                  # [linux]
+        - test -f $PREFIX/targets/{{ target_name }}/lib/libcusolver.so.{{ version.split(".")[0] }}    # [linux]
+        - test -L $PREFIX/lib/libcusolverMg.so.{{ version }}                # [linux]
+        - test -L $PREFIX/lib/libcusolverMg.so.{{ version.split(".")[0] }}  # [linux]
+        - test -L $PREFIX/targets/{{ target_name }}/lib/libcusolverMg.so.{{ version }}                # [linux]
+        - test -f $PREFIX/targets/{{ target_name }}/lib/libcusolverMg.so.{{ version.split(".")[0] }}  # [linux]
+        - if not exist %LIBRARY_BIN%\cusolver*.dll exit 1                  # [win]
+        - if not exist %LIBRARY_BIN%\cusolverMg*.dll exit 1                # [win]
+    about:
+      home: https://developer.nvidia.com/cusolver
+      license_file: LICENSE
+      license: LicenseRef-NVIDIA-End-User-License-Agreement
+      license_url: https://docs.nvidia.com/cuda/eula/index.html
+      summary: CUDA Linear Solver Library
+      description: |
+        cuSOLVER - Direct Linear Solvers on NVIDIA GPUs
+      doc_url: https://docs.nvidia.com/cuda/cusolver/
+
+  - name: libcusolver-dev
+    build:
+      run_exports:
+        - {{ pin_subpackage("libcusolver", max_pin="x") }}
+    files:
+      - lib/libcusolver*.so               # [linux]
+      - lib/pkgconfig                     # [linux]
+      - targets/{{ target_name }}/include              # [linux]
+      - targets/{{ target_name }}/lib/libcusolver*.so  # [linux]
+      - targets/{{ target_name }}/lib/stubs            # [linux]
+      # Not shipping Fortran bindings (in src) on Windows for now
+      - targets/{{ target_name }}/libcusolver/src      # [linux]
+      - Library\include                                # [win]
+      - Library\lib                                    # [win]
+    requirements:
+      build:
+        - {{ compiler("c") }}
+        - {{ compiler("cxx") }}
+        - arm-variant * {{ arm_variant_type }}  # [aarch64]
+        - sysroot_{{ target_platform }} 2.17    # [linux]
+      host:
+        - cuda-version {{ cuda_version }}
+      run:
+        - {{ pin_compatible("cuda-version", max_pin="x.x") }}
+        - {{ pin_subpackage("libcusolver", exact=True) }}
+      run_constrained:
+        - arm-variant * {{ arm_variant_type }}  # [aarch64]
+        - libcusolver-static >={{ version }}      # [linux]
+    test:
+      commands:
+        - test -L $PREFIX/lib/libcusolver.so                                # [linux]
+        - test -L $PREFIX/lib/libcusolverMg.so                              # [linux]
+        - test -f $PREFIX/lib/pkgconfig/cusolver*.pc                        # [linux]
+        - test -f $PREFIX/targets/{{ target_name }}/include/cusolverDn.h                   # [linux]
+        - test -f $PREFIX/targets/{{ target_name }}/include/cusolverMg.h                   # [linux]
+        - test -f $PREFIX/targets/{{ target_name }}/include/cusolverRf.h                   # [linux]
+        - test -f $PREFIX/targets/{{ target_name }}/include/cusolverSp.h                   # [linux]
+        - test -f $PREFIX/targets/{{ target_name }}/include/cusolverSp_LOWLEVEL_PREVIEW.h  # [linux]
+        - test -f $PREFIX/targets/{{ target_name }}/include/cusolver_common.h              # [linux]
+        - test -L $PREFIX/targets/{{ target_name }}/lib/libcusolver.so                     # [linux]
+        - test -L $PREFIX/targets/{{ target_name }}/lib/libcusolverMg.so                   # [linux]
+        - test -f $PREFIX/targets/{{ target_name }}/lib/stubs/libcusolver.so               # [linux]
+        - test -f $PREFIX/targets/{{ target_name }}/lib/stubs/libcusolverMg.so             # [linux]
+        - if not exist %LIBRARY_INC%\cusolverDn.h exit 1                   # [win]
+        - if not exist %LIBRARY_INC%\cusolverMg.h exit 1                   # [win]
+        - if not exist %LIBRARY_INC%\cusolverRf.h exit 1                   # [win]
+        - if not exist %LIBRARY_INC%\cusolverSp.h exit 1                   # [win]
+        - if not exist %LIBRARY_INC%\cusolverSp_LOWLEVEL_PREVIEW.h exit 1  # [win]
+        - if not exist %LIBRARY_INC%\cusolver_common.h exit 1              # [win]
+        - if not exist %LIBRARY_LIB%\cusolver*.lib exit 1                  # [win]
+        - if not exist %LIBRARY_LIB%\cusolverMg*.lib exit 1                # [win]
+    about:
+      home: https://developer.nvidia.com/cusolver
+      license_file: LICENSE
+      license: LicenseRef-NVIDIA-End-User-License-Agreement
+      license_url: https://docs.nvidia.com/cuda/eula/index.html
+      summary: CUDA Linear Solver Library
+      description: |
+        cuSOLVER - Direct Linear Solvers on NVIDIA GPUs
+      doc_url: https://docs.nvidia.com/cuda/cusolver/index.html
+
+  - name: libcusolver-static
+    build:
+      skip: true  # [win]
+    files:
+      - targets/{{ target_name }}/lib/libcusolver*_static.a
+      - targets/{{ target_name }}/lib/libmetis_static.a
+    requirements:
+      build:
+        - {{ compiler("c") }}
+        - {{ compiler("cxx") }}
+        - arm-variant * {{ arm_variant_type }}  # [aarch64]
+        - sysroot_{{ target_platform }} 2.17    # [linux]
+      host:
+        - cuda-version {{ cuda_version }}
+        - libcublas-static
+        - libcusparse-static
+      run:
+        - {{ pin_compatible("cuda-version", max_pin="x.x") }}
+        - {{ pin_compatible("libcublas-static", max_pin="x.x") }}
+        - {{ pin_compatible("libcusparse-static", max_pin="x.x") }}
+      run_constrained:
+        - arm-variant * {{ arm_variant_type }}  # [aarch64]
+    test:
+      commands:
+        - test -f $PREFIX/targets/{{ target_name }}/lib/libcusolver_static.a
+        - test -f $PREFIX/targets/{{ target_name }}/lib/libcusolver_lapack_static.a
+        - test -f $PREFIX/targets/{{ target_name }}/lib/libcusolver_metis_static.a
+        - test -f $PREFIX/targets/{{ target_name }}/lib/libmetis_static.a
+    about:
+      home: https://developer.nvidia.com/cusolver
+      license_file: LICENSE
+      license: LicenseRef-NVIDIA-End-User-License-Agreement
+      license_url: https://docs.nvidia.com/cuda/eula/index.html
+      summary: CUDA Linear Solver Library
+      description: |
+        cuSOLVER - Direct Linear Solvers on NVIDIA GPUs
+      doc_url: https://docs.nvidia.com/cuda/cusolver/index.html
+
+about:
+  home: https://developer.nvidia.com/cusolver
+  license_file: LICENSE
+  license: LicenseRef-NVIDIA-End-User-License-Agreement
+  license_url: https://docs.nvidia.com/cuda/eula/index.html
+  summary: CUDA Linear Solver Library
+  description: |
+    cuSOLVER - Direct Linear Solvers on NVIDIA GPUs
+  doc_url: https://docs.nvidia.com/cuda/cusolver/index.html
+
+extra:
+  recipe-maintainers:
+    - adibbley

recipes/libcusparse/bld.bat

@@ -0,0 +1,5 @@
+if not exist %PREFIX% mkdir %PREFIX%
+
+move lib\x64\* %LIBRARY_LIB%
+move bin\* %LIBRARY_BIN%
+move include\* %LIBRARY_INC%

recipes/libcusparse/build.sh

@@ -0,0 +1,33 @@
+#!/bin/bash
+
+# Install to conda style directories
+[[ -d lib64 ]] && mv lib64 lib
+mkdir -p ${PREFIX}/lib
+[[ -d pkg-config ]] && mv pkg-config ${PREFIX}/lib/pkgconfig
+[[ -d "$PREFIX/lib/pkgconfig" ]] && sed -E -i "s|cudaroot=.+|cudaroot=$PREFIX|g" $PREFIX/lib/pkgconfig/cusparse*.pc
+
+[[ ${target_platform} == "linux-64" ]] && targetsDir="targets/x86_64-linux"
+[[ ${target_platform} == "linux-ppc64le" ]] && targetsDir="targets/ppc64le-linux"
+[[ ${target_platform} == "linux-aarch64" ]] && targetsDir="targets/sbsa-linux"
+
+for i in `ls`; do
+    [[ $i == "build_env_setup.sh" ]] && continue
+    [[ $i == "conda_build.sh" ]] && continue
+    [[ $i == "metadata_conda_debug.yaml" ]] && continue
+    if [[ $i == "lib" ]] || [[ $i == "include" ]]; then
+        # Headers and libraries are installed to targetsDir
+        mkdir -p ${PREFIX}/${targetsDir}
+        mkdir -p ${PREFIX}/$i
+        cp -rv $i ${PREFIX}/${targetsDir}
+        if [[ $i == "lib" ]]; then
+            for j in "$i"/*.so*; do
+                # Shared libraries are symlinked in $PREFIX/lib
+                ln -s ${PREFIX}/${targetsDir}/$j ${PREFIX}/$j
+            done
+        fi
+    else
+        # Put all other files in targetsDir
+        mkdir -p ${PREFIX}/${targetsDir}/${PKG_NAME}
+        cp -rv $i ${PREFIX}/${targetsDir}/${PKG_NAME}
+    fi
+done

recipes/libcusparse/conda_build_config.yaml

@@ -0,0 +1,2 @@
+arm_variant_type:  # [aarch64]
+  - sbsa           # [aarch64]