File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff 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
Original file line number Diff line number Diff 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
You can’t perform that action at this time.
0 commit comments