Home
last modified time | relevance | path

Searched full:cuda_version (Results 1 – 25 of 154) sorted by relevance

1234567

/external/tensorflow/third_party/gpus/
Dfind_cuda_config.py153 def _get_default_cuda_paths(cuda_version): argument
154 if not cuda_version:
155 cuda_version = "*"
156 elif not "." in cuda_version:
157 cuda_version = cuda_version + ".*"
164 cuda_version)
166 return ["/usr/local/cuda-%s" % cuda_version, "/usr/local/cuda", "/usr",
250 version = int(_get_header_version(path, "CUDA_VERSION"))
258 cuda_version = header_version # x.y, see above.
260 cuda_library_path = _find_library(base_paths, "cudart", cuda_version)
[all …]
Dcuda_configure.bzl598 cuda_config.cuda_version,
665 cuda_version: The version of CUDA on the system.
676 cuda_version = config["cuda_version"].split(".")
677 cuda_major = cuda_version[0]
678 cuda_minor = cuda_version[1]
680 cuda_version = ("64_%s%s" if is_windows else "%s.%s") % (cuda_major, cuda_minor)
698 cudart_version = cuda_version
705 cudart_version = cuda_version
706 cublas_version = cuda_version
707 cusolver_version = cuda_version
[all …]
/external/pytorch/.ci/docker/conda/
Dbuild.sh20 CUDA_VERSION=${CUDA_VERSION:-12.1}
22 case ${CUDA_VERSION} in
32 BASE_TARGET=cuda${CUDA_VERSION}
33 DOCKER_TAG=cuda${CUDA_VERSION}
44 --build-arg "CUDA_VERSION=${CUDA_VERSION}" \
56 docker run --rm "${DOCKER_IMAGE_NAME}" nvcc --version | grep "cuda_${CUDA_VERSION}"
DDockerfile1 ARG CUDA_VERSION=10.2
2 ARG BASE_TARGET=cuda${CUDA_VERSION}
52 ARG CUDA_VERSION=10.2
55 ENV CUDA_HOME=/usr/local/cuda-${CUDA_VERSION}
56 # Preserve CUDA_VERSION for the builds
57 ENV CUDA_VERSION=${CUDA_VERSION}
59 ENV PATH=/usr/local/cuda-${CUDA_VERSION}/bin:$PATH
/external/pytorch/.ci/pytorch/win-test-helpers/installation-helpers/
Dinstall_magma.bat1 if "%CUDA_VERSION%" == "cpu" (
6 rem remove dot in cuda_version, fox example 11.1 to 111
12 if x%CUDA_VERSION:.=%==x%CUDA_VERSION% (
13 echo CUDA version %CUDA_VERSION% format isn't correct, which doesn't contain '.'
17 set VERSION_SUFFIX=%CUDA_VERSION:.=%
21 echo unknown CUDA version, please set `CUDA_VERSION` higher than 10.2
/external/pytorch/.github/workflows/
Dbuild-conda-images.yml38 cuda_version: ["11.8", "12.1", "12.4", "cpu"]
40 CUDA_VERSION: ${{ matrix.cuda_version }}
50 …docker-image-name: conda-builder${{ matrix.cuda_version == 'cpu' && '-' || '-cuda' }}${{matrix.cud…
66 …cker/conda/build.sh conda-builder${{ matrix.cuda_version == 'cpu' && ':' || ':cuda' }}${{matrix.cu…
Dbuild-manywheel-images.yml42 cuda_version: ["12.4", "12.1", "11.8"]
45 GPU_ARCH_VERSION: ${{ matrix.cuda_version }}
57 docker-image-name: manylinux-builder-cuda${{matrix.cuda_version}}
73 .ci/docker/manywheel/build.sh manylinux-builder:cuda${{matrix.cuda_version}}
80 cuda_version: ["12.4", "12.1", "11.8"]
83 GPU_ARCH_VERSION: ${{ matrix.cuda_version }}
95 docker-image-name: manylinux2_28-builder-cuda${{matrix.cuda_version}}
111 .ci/docker/manywheel/build.sh manylinux2_28-builder:cuda${{matrix.cuda_version}}
117 cuda_version: ["12.4"]
120 GPU_ARCH_VERSION: ${{ matrix.cuda_version }}
[all …]
/external/pytorch/.ci/docker/
Dbuild.sh95 CUDA_VERSION=12.4.1
109 CUDA_VERSION=12.1.1
123 CUDA_VERSION=12.4.1
138 CUDA_VERSION=12.1.1
153 CUDA_VERSION=12.1.1
168 CUDA_VERSION=12.4.1
183 CUDA_VERSION=11.8.0
197 CUDA_VERSION=12.4.1
211 CUDA_VERSION=12.1.1
225 CUDA_VERSION=12.4.1
[all …]
/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/
Dcuda_stub.cc53 #if CUDA_VERSION < 8000
64 #if CUDA_VERSION < 10000
96 #if CUDA_VERSION < 10000
98 #elif CUDA_VERSION < 10010
100 #elif CUDA_VERSION < 10020
102 #elif CUDA_VERSION < 11000
104 #elif CUDA_VERSION < 11020
Dcublas_stub.cc60 #if CUDA_VERSION < 9000
64 #if CUDA_VERSION < 10000
66 #elif CUDA_VERSION < 10010
68 #elif CUDA_VERSION < 10020
70 #elif CUDA_VERSION < 11000
Dcusparse_stub.cc54 #if CUDA_VERSION < 10000
56 #elif CUDA_VERSION < 10010
58 #elif CUDA_VERSION < 10020
60 #elif CUDA_VERSION < 11000
Dcusolver_stub.cc55 #if CUDA_VERSION < 10000
57 #elif CUDA_VERSION < 10010
59 #elif CUDA_VERSION < 10020
61 #elif CUDA_VERSION < 11000
/external/tensorflow/tensorflow/tools/toolchains/remote_config/
Dconfigs.bzl21 cuda_version = "10.0",
33 cuda_version = "10.1",
46 cuda_version = "10.1",
59 cuda_version = "10.2",
72 cuda_version = "11.0",
88 cuda_version = "10.1",
99 cuda_version = "11.0",
132 cuda_version = "11.2",
146 cuda_version = "11.2",
158 cuda_version = "11.1",
[all …]
Drbe_config.bzl15 def _tensorflow_rbe_config(name, compiler, python_versions, os, rocm_version = None, cuda_version =…
16 if cuda_version != None and rocm_version != None:
17 fail("Specifying both cuda_version and rocm_version is not supported.")
34 if cuda_version != None:
43 "TF_CUDA_VERSION": cuda_version,
54 container_name = "cuda%s-cudnn%s-%s" % (cuda_version, cudnn_version, os)
106 fail("Neither cuda_version, rocm_version nor python_version specified.")
/external/pytorch/docs/source/notes/
Dhip.rst129 NOTE: The CUDA_VERSION macro, cudaRuntimeGetVersion and cudaDriverGetVersion APIs do not
135 ``#if defined(CUDA_VERSION) && CUDA_VERSION >= 11000`` to implicitly exclude ROCm/HIP,
139 ``#if defined(CUDA_VERSION) && CUDA_VERSION >= 11000 && !defined(USE_ROCM)``
143 ``#if (defined(CUDA_VERSION) && CUDA_VERSION >= 11000) || defined(USE_ROCM)``
147 ``#if (defined(CUDA_VERSION) && CUDA_VERSION >= 11000) || (defined(USE_ROCM) && ROCM_VERSION >= 403…
/external/pytorch/.ci/pytorch/win-test-helpers/
Dbuild_pytorch.bat67 set CUDA_PATH=C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v%CUDA_VERSION%
69 if x%CUDA_VERSION:.=%==x%CUDA_VERSION% (
70 echo CUDA version %CUDA_VERSION% format isn't correct, which doesn't contain '.'
74 if x%CUDA_VERSION:.=%==x%CUDA_VERSION% (
75 echo CUDA version %CUDA_VERSION% format isn't correct, which doesn't contain '.'
78 set VERSION_SUFFIX=%CUDA_VERSION:.=_%
/external/pytorch/torch/_inductor/codegen/cuda/
Dcuda_env.py28 cuda_version = config.cuda.version
29 if cuda_version is None:
30 cuda_version = torch.version.cuda
31 return cuda_version
/external/pytorch/aten/src/ATen/native/sparse/cuda/
DSparseSemiSturcturedApply.cu8 #if defined(USE_ROCM) || defined(_MSC_VER) || (defined(CUDA_VERSION) && CUDA_VERSION < 11080)
15 #if defined(USE_ROCM) || defined(_MSC_VER) || (defined(CUDA_VERSION) && CUDA_VERSION < 11080)
92 #if defined(USE_ROCM) || defined(_MSC_VER) || (defined(CUDA_VERSION) && CUDA_VERSION < 11080) in _sparse_semi_structured_apply()
DSparseSemiStructuredApplyDense.cu7 #if defined(USE_ROCM) || defined(_MSC_VER) || (defined(CUDA_VERSION) && CUDA_VERSION < 11080)
15 #if defined(USE_ROCM) || defined(_MSC_VER) || (defined(CUDA_VERSION) && CUDA_VERSION < 11080)
126 #if defined(USE_ROCM) || defined(_MSC_VER) || (defined(CUDA_VERSION) && CUDA_VERSION < 11080) in _sparse_semi_structured_apply_dense()
/external/pytorch/tools/
Dgenerate_torch_version.py77 parser.add_argument("--cuda-version", "--cuda_version", type=str)
83 args.cuda_version = None if args.cuda_version == "" else args.cuda_version
104 f.write(f"cuda: Optional[str] = {repr(args.cuda_version)}\n")
/external/pytorch/.ci/docker/ubuntu-cuda/
DDockerfile2 ARG CUDA_VERSION
8 ARG CUDA_VERSION
131 RUN if [ -n "${CUDA_VERSION}" ]; then bash install_openmpi.sh; fi
148 ARG CUDA_VERSION
154 ARG CUDA_VERSION
160 ARG CUDA_VERSION
/external/pytorch/cmake/Modules_CUDA_fix/upstream/FindCUDA/
Dselect_compute_arch.cmake23 set(CUDA_VERSION "${CMAKE_MATCH_1}") variable
38 if(CUDA_VERSION VERSION_GREATER "10.5")
43 if(CUDA_VERSION VERSION_LESS "11.1")
49 if(NOT CUDA_VERSION VERSION_LESS "11.1")
54 if(CUDA_VERSION VERSION_LESS "11.8")
60 if(NOT CUDA_VERSION VERSION_LESS "11.8")
68 if(CUDA_VERSION VERSION_LESS "12.0")
75 if(NOT CUDA_VERSION VERSION_LESS "12.0")
/external/pytorch/.ci/docker/common/
Dinstall_cudnn.sh7 if [[ ${CUDA_VERSION:0:2} == "12" ]]; then
9 elif [[ ${CUDA_VERSION:0:2} == "11" ]]; then
12 print "Unsupported CUDA version ${CUDA_VERSION}"
/external/pytorch/torch/csrc/cuda/shared/
Dcudart.cpp31 #if !defined(USE_ROCM) && defined(CUDA_VERSION) && CUDA_VERSION < 12000 in initCudartBindings()
99 #if !defined(USE_ROCM) && defined(CUDA_VERSION) && CUDA_VERSION < 12000 in initCudartBindings()
/external/sdv/vsomeip/third_party/boost/predef/include/boost/predef/language/
Dcuda.h27 | `CUDA_VERSION` | VV.RR.P
36 # if defined(CUDA_VERSION)
37 # define BOOST_LANG_CUDA BOOST_PREDEF_MAKE_10_VVRRP(CUDA_VERSION)

1234567