Skip to content
Draft
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
26 changes: 8 additions & 18 deletions recipe/meta.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -94,10 +94,17 @@ outputs:
- echo "nvcc is %NVCC%, CUDA path is %CUDA_PATH%" # [win]
# Workaround __ieee128 error; see https://github.com/LLNL/blt/issues/341
- export NVCC="$NVCC -Xcompiler -mno-float128" # [ppc64le]


# When cross-compiling, symlink `targets` from `$PREFIX` to `$BUILD_PREFIX`
- ln -s "${PREFIX}/targets/ppc64le-linux" "${BUILD_PREFIX}/targets/ppc64le-linux" # [ppc64le and (build_platform != target_platform)]
- ln -s "${PREFIX}/targets/sbsa-linux" "${BUILD_PREFIX}/targets/sbsa-linux" # [aarch64 and (build_platform != target_platform)]

- >-
{{ PYTHON }} -m pip install . --no-deps -vv
- if errorlevel 1 exit 1 # [win]

- rm "${PREFIX}/targets/ppc64le-linux" # [ppc64le and (build_platform != target_platform)]
- rm "${PREFIX}/targets/sbsa-linux" # [aarch64 and (build_platform != target_platform)]

# copy activate/deactivate scripts
- mkdir -p "${PREFIX}/etc/conda/activate.d" # [linux]
Expand Down Expand Up @@ -149,23 +156,6 @@ outputs:
- cross-python_{{ target_platform }} # [build_platform != target_platform]
- python # [build_platform != target_platform]
- cython >=0.29.22,<3 # [build_platform != target_platform]
# TODO: clean up
{% if cuda_major >= 12 %}
- cuda-driver-dev # [build_platform != target_platform and linux]
- cuda-cudart-dev # [build_platform != target_platform]
- cuda-nvrtc-dev # [build_platform != target_platform]
- cuda-nvtx-dev # [build_platform != target_platform]
- cuda-profiler-api # [build_platform != target_platform]
- libcublas-dev # [build_platform != target_platform]
- libcufft-dev # [build_platform != target_platform]
- libcurand-dev # [build_platform != target_platform]
- libcusolver-dev # [build_platform != target_platform]
- libcusparse-dev # [build_platform != target_platform]
{% endif %}
# optional dependencies for CUDA 11.2+
- cudnn ~=8.8 # [build_platform != target_platform and not (aarch64 or (ppc64le and (cuda_compiler_version or "").startswith("12")))]
- nccl ~=2.16 # [build_platform != target_platform]
- cutensor ~=2.0 # [build_platform != target_platform]
host:
- python
- pip
Expand Down