Home
last modified time | relevance | path

Searched full:rocm_version (Results 1 – 25 of 33) sorted by relevance

12

/external/pytorch/.ci/docker/common/
Dinstall_miopen.sh6 ROCM_VERSION=$1
8 if [[ -z $ROCM_VERSION ]]; then
9 echo "missing ROCM_VERSION"
30 IFS=. ROCM_VERSION_ARRAY=(${ROCM_VERSION})
41 echo "Unhandled ROCM_VERSION ${ROCM_VERSION}"
59 ## Need a sanitized ROCM_VERSION without patchlevel; patchlevel version 0 must be added to paths.
60 ROCM_DOTS=$(echo ${ROCM_VERSION} | tr -d -c '.' | wc -c)
62 ROCM_VERSION_NOPATCH="${ROCM_VERSION}"
63 ROCM_INSTALL_PATH="/opt/rocm-${ROCM_VERSION}.0"
65 ROCM_VERSION_NOPATCH="${ROCM_VERSION%.*}"
[all …]
Dinstall_rocm.sh28 …echo "deb [arch=amd64] https://repo.radeon.com/amdgpu/${ROCM_VERSION}/ubuntu ${UBUNTU_VERSION_NAME…
32 local rocm_baseurl="http://repo.radeon.com/rocm/apt/${ROCM_VERSION}"
45 if [[ $(ver $ROCM_VERSION) -ge $(ver 6.1) ]]; then
83 amdgpu_baseurl="https://repo.radeon.com/amdgpu/${ROCM_VERSION}/rhel/9.0/main/x86_64"
85 amdgpu_baseurl="https://repo.radeon.com/amdgpu/${ROCM_VERSION}/rhel/7.9/main/x86_64"
94 local rocm_baseurl="http://repo.radeon.com/rocm/yum/${ROCM_VERSION}"
Dinstall_cache.sh35 if [ -n "$ROCM_VERSION" ]; then
59 # NOTE: See specific ROCM_VERSION case below.
60 if [ "x$ROCM_VERSION" = x ]; then
76 if [ -n "$ROCM_VERSION" ]; then
/external/pytorch/.github/scripts/amd/
Dpackage_triton_wheel.sh6 # Find rocm_version.h header file for ROCm version extract
7 rocm_version_h="${ROCM_HOME}/include/rocm-core/rocm_version.h"
9 rocm_version_h="${ROCM_HOME}/include/rocm_version.h"
12 # Error out if rocm_version.h not found
14 echo "Error: rocm_version.h not found in expected locations." >&2
/external/tensorflow/tensorflow/tools/toolchains/remote_config/
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.")
78 elif rocm_version != None:
106 fail("Neither cuda_version, rocm_version nor python_version specified.")
Dconfigs.bzl113 rocm_version = "5.0", # Any version will do.
123 rocm_version = "5.0", # Any version will do.
/external/pytorch/torch/testing/_internal/
Dcommon_cuda.py225 rocm_version = str(torch.version.hip)
226 rocm_version = rocm_version.split("-")[0] # ignore git sha
227 return tuple(int(x) for x in rocm_version.split("."))
236 rocm_version = str(torch.version.hip)
237 rocm_version = rocm_version.split("-")[0] # ignore git sha
238 rocm_version_tuple = tuple(int(x) for x in rocm_version.split("."))
/external/pytorch/cmake/public/
DLoadHIP.cmake52 if(EXISTS ${ROCM_INCLUDE_DIRS}/rocm_version.h)
55 "#include <rocm_version.h>\n"
57 elseif(EXISTS ${ROCM_INCLUDE_DIRS}/rocm-core/rocm_version.h)
60 "#include <rocm-core/rocm_version.h>\n"
63 message("********************* rocm_version.h couldnt be found ******************\n")
92 message("\n***** ROCm version from rocm_version.h ****\n")
/external/pytorch/.github/workflows/
Dbuild-libtorch-images.yml72 rocm_version: ["6.1", "6.2"]
75 GPU_ARCH_VERSION: ${{ matrix.rocm_version }}
85 docker-image-name: libtorch-cxx11-builder-rocm${{matrix.rocm_version}}
101 .ci/docker/libtorch/build.sh libtorch-cxx11-builder:rocm${{matrix.rocm_version}}
Dbuild-manywheel-images.yml150 rocm_version: ["6.1", "6.2"]
153 GPU_ARCH_VERSION: ${{ matrix.rocm_version }}
163 docker-image-name: manylinux-builder-rocm${{matrix.rocm_version}}
180 .ci/docker/manywheel/build.sh manylinux-builder:rocm${{matrix.rocm_version}}
/external/pytorch/.ci/docker/manywheel/
DDockerfile_2_282 ARG ROCM_VERSION=3.7
135 ARG ROCM_VERSION=3.7
138 RUN bash ./install_rocm.sh ${ROCM_VERSION} && rm install_rocm.sh
145 RUN bash ./install_miopen.sh ${ROCM_VERSION} && rm install_miopen.sh
DDockerfile2 ARG ROCM_VERSION=3.7
177 ARG ROCM_VERSION=3.7
187 #RUN ROCM_VERSION=${ROCM_VERSION} bash ./install_rocm.sh && rm install_rocm.sh
195 RUN bash ./install_miopen.sh ${ROCM_VERSION} && rm install_miopen.sh
DDockerfile_20142 ARG ROCM_VERSION=3.7
143 ARG ROCM_VERSION=3.7
146 RUN bash ./install_rocm.sh ${ROCM_VERSION} && rm install_rocm.sh
153 RUN bash ./install_miopen.sh ${ROCM_VERSION} && rm install_miopen.sh
Dbuild.sh105 …DOCKER_GPU_BUILD_ARG="--build-arg ROCM_VERSION=${GPU_ARCH_VERSION} --build-arg PYTORCH_ROCM_ARCH=$…
/external/pytorch/caffe2/core/
Dmacros.h.in44 {"ROCM_VERSION", "${ROCM_VERSION}"}, \
/external/pytorch/aten/src/ATen/cuda/tunable/
DTunable.cpp39 #include <rocm-core/rocm_version.h>
187 std::string rocm_version = ROCM_BUILD_INFO; in TuningResultsValidator() local
189 "ROCM_VERSION", in TuningResultsValidator()
190 [rocm_version]() { return rocm_version; }, in TuningResultsValidator()
191 [rocm_version](auto&& k) { return rocm_version == k ? OK : FAIL; }); in TuningResultsValidator()
DREADME.md34 Validator,ROCM_VERSION,6.0.0.0-12969-1544e39
/external/pytorch/.ci/docker/
Dbuild.sh294 ROCM_VERSION=6.1
305 ROCM_VERSION=6.2
440 extract_version_from_image_name rocm ROCM_VERSION
505 --build-arg "ROCM_VERSION=${ROCM_VERSION:-}" \
/external/pytorch/.ci/docker/ubuntu-rocm/
DDockerfile64 ARG ROCM_VERSION
72 RUN bash ./install_miopen.sh ${ROCM_VERSION} && rm install_miopen.sh
/external/pytorch/cmake/
DSummary.cmake130 message(STATUS " ROCM_VERSION : ${ROCM_VERSION}")
/external/pytorch/.ci/docker/centos-rocm/
DDockerfile73 ARG ROCM_VERSION
/external/pytorch/aten/src/ATen/native/cuda/
Dint4mm.cu1 #if (defined(USE_ROCM) && ROCM_VERSION >= 50700) || ((defined(CUDA_VERSION) && CUDA_VERSION >= 1200…
160 #if (defined(USE_ROCM) && ROCM_VERSION >= 50700) || ((defined(CUDA_VERSION) && CUDA_VERSION >= 1200…
282 #if ROCM_VERSION >= 60200 in convert_i4x8_to_bf16x2x4()
1173 #if (defined(USE_ROCM) && ROCM_VERSION >= 50700) || ((defined(CUDA_VERSION) && CUDA_VERSION >= 1200… in _weight_int4pack_mm_cuda()
1339 #if (defined(USE_ROCM) && ROCM_VERSION >= 50700) || ((defined(CUDA_VERSION) && CUDA_VERSION >= 1200… in _convert_weight_to_int4pack_cuda()
/external/pytorch/docs/source/notes/
Dhip.rst147 ``#if (defined(CUDA_VERSION) && CUDA_VERSION >= 11000) || (defined(USE_ROCM) && ROCM_VERSION >= 403…
/external/pytorch/torch/cuda/
Dtunable.py46 Validator,ROCM_VERSION,6.0.0.0-12969-1544e39
/external/tensorflow/third_party/gpus/
Dfind_rocm_config.py87 version_file = os.path.join(path, "include/rocm_version.h")

12