Home
last modified time | relevance | path

Searched full:desired_cuda (Results 1 – 25 of 35) sorted by relevance

12

/external/pytorch/.github/workflows/
Dgenerated-linux-binary-conda-nightly.yml49 DESIRED_CUDA: cpu
67 DESIRED_CUDA: cpu
88 DESIRED_CUDA: cpu
108 DESIRED_CUDA: cu118
128 DESIRED_CUDA: cu118
150 DESIRED_CUDA: cu118
171 DESIRED_CUDA: cu121
191 DESIRED_CUDA: cu121
213 DESIRED_CUDA: cu121
234 DESIRED_CUDA: cu124
[all …]
Dgenerated-linux-binary-manywheel-nightly.yml49 DESIRED_CUDA: cpu
67 DESIRED_CUDA: cpu
88 DESIRED_CUDA: cpu
108 DESIRED_CUDA: cpu-cxx11-abi
127 DESIRED_CUDA: cpu-cxx11-abi
149 DESIRED_CUDA: cpu-cxx11-abi
170 DESIRED_CUDA: cu118
190 DESIRED_CUDA: cu118
212 DESIRED_CUDA: cu118
233 DESIRED_CUDA: cu121
[all …]
Dgenerated-linux-binary-manywheel-split-nightly58 DESIRED_CUDA: cu118
82 DESIRED_CUDA: cu118
106 DESIRED_CUDA: cu118
129 DESIRED_CUDA: cu121
153 DESIRED_CUDA: cu121
177 DESIRED_CUDA: cu121
200 DESIRED_CUDA: cu124
224 DESIRED_CUDA: cu124
248 DESIRED_CUDA: cu124
271 DESIRED_CUDA: cpu
[all …]
Dgenerated-linux-aarch64-binary-manywheel-nightly.yml49 DESIRED_CUDA: cpu
70 DESIRED_CUDA: cpu
92 DESIRED_CUDA: cpu
112 DESIRED_CUDA: cu124
136 DESIRED_CUDA: cu124
157 DESIRED_CUDA: cpu
178 DESIRED_CUDA: cpu
200 DESIRED_CUDA: cpu
220 DESIRED_CUDA: cu124
244 DESIRED_CUDA: cu124
[all …]
Dgenerated-linux-binary-libtorch-pre-cxx11-nightly.yml49 DESIRED_CUDA: cpu
68 DESIRED_CUDA: cpu
90 DESIRED_CUDA: cpu
111 DESIRED_CUDA: cu118
131 DESIRED_CUDA: cu118
154 DESIRED_CUDA: cu118
176 DESIRED_CUDA: cu121
196 DESIRED_CUDA: cu121
219 DESIRED_CUDA: cu121
241 DESIRED_CUDA: cu124
[all …]
Dgenerated-linux-binary-libtorch-cxx11-abi-nightly.yml49 DESIRED_CUDA: cpu
68 DESIRED_CUDA: cpu
90 DESIRED_CUDA: cpu
111 DESIRED_CUDA: cu118
131 DESIRED_CUDA: cu118
154 DESIRED_CUDA: cu118
176 DESIRED_CUDA: cu121
196 DESIRED_CUDA: cu121
219 DESIRED_CUDA: cu121
241 DESIRED_CUDA: cu124
[all …]
Dgenerated-linux-s390x-binary-manywheel-nightly.yml49 DESIRED_CUDA: cpu
70 DESIRED_CUDA: cpu
92 DESIRED_CUDA: cpu
112 DESIRED_CUDA: cpu
133 DESIRED_CUDA: cpu
155 DESIRED_CUDA: cpu
175 DESIRED_CUDA: cpu
196 DESIRED_CUDA: cpu
218 DESIRED_CUDA: cpu
238 DESIRED_CUDA: cpu
[all …]
Dgenerated-linux-binary-manywheel-main.yml44 DESIRED_CUDA: cu118
64 DESIRED_CUDA: cu118
84 DESIRED_CUDA: cu121
104 DESIRED_CUDA: cu121
124 DESIRED_CUDA: cu124
144 DESIRED_CUDA: cu124
Dgenerated-linux-binary-manywheel-split-main53 DESIRED_CUDA: cu118
77 DESIRED_CUDA: cu118
100 DESIRED_CUDA: cu121
124 DESIRED_CUDA: cu121
147 DESIRED_CUDA: cu124
171 DESIRED_CUDA: cu124
D_binary-build-linux.yml41 DESIRED_CUDA:
93 DESIRED_CUDA: ${{ inputs.DESIRED_CUDA }}
121 echo "DESIRED_CUDA=${{ env.DESIRED_CUDA }}"
236 -e DESIRED_CUDA \
D_binary-test-linux.yml30 DESIRED_CUDA:
81 DESIRED_CUDA: ${{ inputs.DESIRED_CUDA }}
109 echo "DESIRED_CUDA=${{ env.DESIRED_CUDA }}"
D_binary-upload.yml26 DESIRED_CUDA:
81 DESIRED_CUDA: ${{ inputs.DESIRED_CUDA }}
145 UPLOAD_SUBFOLDER: "${{ env.DESIRED_CUDA }}"
Dgenerated-macos-arm64-binary-conda-nightly.yml45 DESIRED_CUDA: cpu
141 DESIRED_CUDA: cpu
162 DESIRED_CUDA: cpu
258 DESIRED_CUDA: cpu
279 DESIRED_CUDA: cpu
375 DESIRED_CUDA: cpu
396 DESIRED_CUDA: cpu
492 DESIRED_CUDA: cpu
513 DESIRED_CUDA: cpu
609 DESIRED_CUDA: cpu
Dgenerated-windows-binary-conda-nightly.yml45 DESIRED_CUDA: cpu
156 DESIRED_CUDA: cpu
266 DESIRED_CUDA: cpu
285 DESIRED_CUDA: cu118
397 DESIRED_CUDA: cu118
508 DESIRED_CUDA: cu118
528 DESIRED_CUDA: cu121
640 DESIRED_CUDA: cu121
751 DESIRED_CUDA: cu121
771 DESIRED_CUDA: cu124
[all …]
Dgenerated-windows-binary-wheel-nightly.yml45 DESIRED_CUDA: cpu
157 DESIRED_CUDA: cpu
267 DESIRED_CUDA: cpu
286 DESIRED_CUDA: cu118
399 DESIRED_CUDA: cu118
510 DESIRED_CUDA: cu118
530 DESIRED_CUDA: cu121
643 DESIRED_CUDA: cu121
754 DESIRED_CUDA: cu121
774 DESIRED_CUDA: cu124
[all …]
Dgenerated-macos-arm64-binary-wheel-nightly.yml45 DESIRED_CUDA: cpu
142 DESIRED_CUDA: cpu
163 DESIRED_CUDA: cpu
260 DESIRED_CUDA: cpu
281 DESIRED_CUDA: cpu
378 DESIRED_CUDA: cpu
399 DESIRED_CUDA: cpu
496 DESIRED_CUDA: cpu
517 DESIRED_CUDA: cpu
614 DESIRED_CUDA: cpu
Dgenerated-linux-binary-libtorch-pre-cxx11-main.yml44 DESIRED_CUDA: cpu
63 DESIRED_CUDA: cpu
Dgenerated-linux-binary-libtorch-cxx11-abi-main.yml44 DESIRED_CUDA: cpu
63 DESIRED_CUDA: cpu
/external/pytorch/.circleci/scripts/
Dbinary_linux_build.sh14 if [[ "${DESIRED_CUDA}" =~ cu1[1-2][0-9] ]]; then
21 elif [[ "$DESIRED_CUDA" == cpu ]]; then
23 elif [[ "$DESIRED_CUDA" == *"rocm"* ]]; then
Dbinary_populate_env.sh35 elif [[ "$DESIRED_CUDA" == cpu ]]; then
38 export DOCKER_IMAGE="pytorch/manylinux-cuda${DESIRED_CUDA:2}"
44 if [[ ${DESIRED_CUDA} == "cpu" ]]; then
69 export PYTORCH_BUILD_VERSION="${BASE_BUILD_VERSION}+$DESIRED_CUDA"
135 export DESIRED_CUDA="$DESIRED_CUDA"
Dbinary_linux_test.sh87 if [[ "$DESIRED_CUDA" == 'cpu' ]]; then
90 cu_ver="${DESIRED_CUDA:2:2}.${DESIRED_CUDA:4}"
102 …" "\$pkg_torch" --index-url "https://download.pytorch.org/whl/\${CHANNEL}/${DESIRED_CUDA}_pypi_pkg"
105 pip install "\$pkg" --index-url "https://download.pytorch.org/whl/\${CHANNEL}/${DESIRED_CUDA}"
Dbinary_windows_test.sh6 export CUDA_VERSION="${DESIRED_CUDA/cu/}"
9 if [[ "$DESIRED_CUDA" == 'xpu' ]]; then
Dbinary_windows_build.sh7 export CUDA_VERSION="${DESIRED_CUDA/cu/}"
13 if [[ "$DESIRED_CUDA" == 'xpu' ]]; then
/external/pytorch/.ci/docker/conda/
DDockerfile63 ENV DESIRED_CUDA=11.8
67 ENV DESIRED_CUDA=12.1
71 ENV DESIRED_CUDA=12.4
/external/pytorch/.github/scripts/
Dgenerate_binary_build_matrix.py247 "desired_cuda": translate_desired_cuda(
297 "desired_cuda": translate_desired_cuda(
371 "desired_cuda": translate_desired_cuda(
400 "desired_cuda": translate_desired_cuda(
418 "desired_cuda": translate_desired_cuda(

12