Skip to content

Commit 4a23411

Browse files
Disable cufile for ppc64le
1 parent 4880c02 commit 4a23411

2 files changed

Lines changed: 5 additions & 3 deletions

File tree

recipe/build.sh

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -147,7 +147,9 @@ elif [[ ${cuda_compiler_version} != "None" ]]; then
147147
# with no NVIDIA GPUs.
148148
export USE_MKLDNN=1
149149
export USE_CUDA=1
150-
export USE_CUFILE=1
150+
if [[ "$target_platform" != "linux-ppc64le" ]]; then
151+
export USE_CUFILE=1
152+
fi
151153
# PyTorch has multiple different bits of logic finding CUDA, override
152154
# all of them.
153155
export CUDAToolkit_BIN_DIR=${BUILD_PREFIX}/bin

recipe/meta.yaml

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -101,7 +101,7 @@ requirements:
101101
- cuda-nvml-dev
102102
- cuda-profiler-api
103103
- libcublas-dev
104-
- libcufile-dev
104+
- libcufile-dev # [not ppc64le]
105105
- libcufft-dev
106106
- libcurand-dev
107107
- libcusolver-dev
@@ -218,7 +218,7 @@ outputs:
218218
- cuda-nvml-dev
219219
- cuda-profiler-api
220220
- libcublas-dev
221-
- libcufile-dev
221+
- libcufile-dev # [not ppc64le]
222222
- libcufft-dev
223223
- libcurand-dev
224224
- libcusolver-dev

0 commit comments

Comments
 (0)