| /external/tensorflow/third_party/gpus/rocm/ |
| D | BUILD.tpl | 17 "rocm/rocm_config.h",
22 "rocm/include",
23 "rocm/include/rocrand",
24 "rocm/include/roctracer",
31 srcs = ["rocm/lib/%{hip_lib}"],
32 data = ["rocm/lib/%{hip_lib}"],
35 "rocm/include",
43 srcs = ["rocm/lib/%{rocblas_lib}"],
44 data = ["rocm/lib/%{rocblas_lib}"],
47 "rocm/include",
[all …]
|
| D | build_defs.bzl.tpl | 1 # Macros for building ROCm code.
3 """Shorthand for select()'ing on whether we're building with ROCm.
6 with ROCm enabled. Otherwise, the select statement evaluates to if_false.
10 "@local_config_rocm//rocm:using_hipcc": if_true,
16 """Default options for all ROCm compilations."""
17 return if_rocm(["-x", "rocm"] + %{rocm_extra_copts})
20 """Gets the appropriate set of copts for (maybe) ROCm compilation.
22 If we're doing ROCm compilation, returns copts for our particular ROCm
23 compiler. If we're not doing ROCm compilation, returns an empty list.
28 "@local_config_rocm//rocm:using_hipcc": ([
[all …]
|
| /external/pytorch/.ci/docker/common/ |
| D | install_rocm.sh | 30 # Add rocm repository
31 wget -qO - http://repo.radeon.com/rocm/rocm.gpg.key | apt-key add -
32 local rocm_baseurl="http://repo.radeon.com/rocm/apt/${ROCM_VERSION}"
33 … "deb [arch=amd64] ${rocm_baseurl} ${UBUNTU_VERSION_NAME} main" > /etc/apt/sources.list.d/rocm.list
37 rocm-dev \
38 rocm-utils \
39 rocm-libs \
46 DEBIAN_FRONTEND=noninteractive apt-get install -y --allow-unauthenticated rocm-llvm-dev
49 # precompiled miopen kernels added in ROCm 3.5, renamed in ROCm 5.5
59 …# ROCm 6.0 had a regression where journal_mode was enabled on the kdb files resulting in permissio…
[all …]
|
| D | install_cache.sh | 77 # ROCm compiler is hcc or clang. However, it is commonly invoked via hipcc wrapper.
78 # hipcc will call either hcc or clang using an absolute path starting with /opt/rocm,
80 # directly under /opt/rocm while also preserving the original compiler names.
95 if [[ -e "/opt/rocm/hcc/bin/hcc" ]]; then
96 # ROCm 3.3 or earlier.
97 mkdir /opt/rocm/hcc/bin/original
98 write_sccache_stub_rocm /opt/rocm/hcc/bin/hcc
99 write_sccache_stub_rocm /opt/rocm/hcc/bin/clang
100 write_sccache_stub_rocm /opt/rocm/hcc/bin/clang++
102 pushd /opt/rocm/hcc/bin/original
[all …]
|
| D | install_miopen.sh | 46 # Install custom MIOpen + COMgr for ROCm >= 4.0.1
48 echo "ROCm version < 4.0.1; will not install custom MIOpen"
63 ROCM_INSTALL_PATH="/opt/rocm-${ROCM_VERSION}.0"
66 ROCM_INSTALL_PATH="/opt/rocm-${ROCM_VERSION}"
74 # Pull MIOpen repo and set DMIOPEN_EMBED_DB based on ROCm version
76 MIOPEN_BRANCH="release/rocm-rel-6.2-staging"
78 echo "ROCm 6.1 MIOpen does not need any patches, do not build from source"
81 echo "ROCm 6.0 MIOpen does not need any patches, do not build from source"
84 echo "ROCm 5.7 MIOpen does not need any patches, do not build from source"
87 MIOPEN_BRANCH="release/rocm-rel-5.6-staging"
[all …]
|
| D | install_rocm_magma.sh | 9 # "install" hipMAGMA into /opt/rocm/magma by copying after build
13 # Version 2.7.2 + ROCm related updates
21 …'LIB += -Wl,--enable-new-dtags -Wl,--rpath,/opt/rocm/lib -Wl,--rpath,$(MKLROOT)/lib -Wl,--rpath,/o…
23 export PATH="${PATH}:/opt/rocm/bin"
38 mv magma /opt/rocm
|
| /external/tensorflow/third_party/gpus/ |
| D | rocm_configure.bzl | 1 """Repository rule for ROCm autoconfiguration.
5 * `TF_NEED_ROCM`: Whether to enable building with ROCm.
7 * `ROCM_PATH`: The path to the ROCm toolkit. Default is `/opt/rocm`.
38 _DEFAULT_ROCM_TOOLKIT_PATH = "/opt/rocm"
41 """Verify all variables that crosstool/BUILD.rocm.tpl expects are substituted.
60 "BUILD.rocm.tpl template is missing these variables: " +
150 """Output failure message when rocm configuration fails."""
164 """Generates the cxx_builtin_include_directory entries for rocm inc dirs.
193 # Support hcc based off clang 10.0.0 (for ROCm 3.3)
206 auto_configure_warning("ROCm configure is only supported on Linux")
[all …]
|
| D | find_rocm_config.py | 15 """Prints ROCm library and header directories and versions found on the system.
17 The script searches for ROCm library and header files on the system, inspects
26 specified, then "/opt/rocm" is used as it default value
41 return "/opt/rocm"
45 """Determines and returns the ROCm installation path."""
75 raise ConfigError("{} ROCm version file ".format(prior_err) +
89 return False, 'ROCm version file "{}" not found.'.format(version_file) +\
90 " Trying an alternate approach to determine the ROCm version.", 0,0,0
158 "rocblas/include/rocblas-version.h", # ROCm 3.7 and prior
159 "rocblas/include/internal/rocblas-version.h", # ROCm 3.8
[all …]
|
| /external/pytorch/torch/_inductor/codegen/rocm/ |
| D | compile_command.py | 17 os.path.join(config.rocm.rocm_home, "include")
18 if config.rocm.rocm_home
21 if not config.rocm.ck_dir:
24 config.rocm.ck_dir or cpp_extension._join_rocm_home("composable_kernel"),
34 os.path.join(config.rocm.rocm_home, "lib")
35 if config.rocm.rocm_home
39 os.path.join(config.rocm.rocm_home, "hip", "lib")
40 if config.rocm.rocm_home
52 arch_list = config.rocm.arch or ["native"]
55 config.rocm.compile_opt_level,
[all …]
|
| /external/tensorflow/tensorflow/tools/ci_build/ |
| D | Dockerfile.rbe.rocm-ubuntu20.04-manylinux2014-multipython | 1 # Dockerfile for ROCm RBE builds.
6 # $ docker build -f Dockerfile.rbe.rocm-ubuntu20.04-manylinux2014-multipython \
7 # --tag "gcr.io/tensorflow-testing/nosla-rocm-ubuntu20.04-manylinux2014-multipython" .
8 # $ docker push gcr.io/tensorflow-testing/nosla-rocm-ubuntu20.04-manylinux2014-multipython
16 # Install ROCm packages
19 && curl -k -sL https://repo.radeon.com/rocm/rocm.gpg.key | apt-key add - \
20 && printf "deb [arch=amd64] http://repo.radeon.com/rocm/apt/5.0/ ubuntu main" \
21 | tee /etc/apt/sources.list.d/rocm.list \
23 rocm-dev rocm-libs rccl \
26 # Set ROCm environment variables and paths.
[all …]
|
| D | Dockerfile.rbe.rocm-ubuntu18.04-manylinux2010-multipython | 1 # Dockerfile for ROCm RBE builds.
6 # $ docker build -f Dockerfile.rbe.rocm-ubuntu18.04-manylinux2010-multipython \
7 # --tag "gcr.io/tensorflow-testing/nosla-rocm-ubuntu18.04-manylinux2010-multipython" .
8 # $ docker push gcr.io/tensorflow-testing/nosla-rocm-ubuntu18.04-manylinux2010-multipython
15 # Install ROCm packages
18 && curl -k -sL https://repo.radeon.com/rocm/rocm.gpg.key | apt-key add - \
19 …intf "deb [arch=amd64] http://repo.radeon.com/rocm/apt/5.0/ ubuntu main" | tee /etc/apt/sources.li…
21 rocm-dev rocm-libs rccl \
24 # Set ROCm environment variables and paths.
25 # We use /opt/rocm without version suffix so that the toolchain configuration
[all …]
|
| D | Dockerfile.rocm | 1 # This Dockerfile provides a starting point for a ROCm installation of
6 ARG ROCM_DEB_REPO=https://repo.radeon.com/rocm/apt/5.1/
9 ARG ROCM_PATH=/opt/rocm-5.1.0
16 # Add rocm repository
18 RUN wget -qO - https://repo.radeon.com/rocm/rocm.gpg.key | apt-key add -;
19 RUN bin/bash -c 'if [[ $ROCM_DEB_REPO == https://repo.radeon.com/rocm/* ]] ; then \
20 …arch=amd64] $ROCM_DEB_REPO $ROCM_BUILD_NAME $ROCM_BUILD_NUM" > /etc/apt/sources.list.d/rocm.list; \
22 …usted=yes] $ROCM_DEB_REPO $ROCM_BUILD_NAME $ROCM_BUILD_NUM" > /etc/apt/sources.list.d/rocm.list ; \
67 # Install rocm pkgs
70 rocm-dev rocm-libs rccl && \
[all …]
|
| /external/pytorch/test/inductor/ |
| D | test_ck_backend.py | 61 @unittest.skipIf(not torch.version.hip, "ROCM only")
83 assert "rocm" in dir(config)
91 "rocm.n_max_profiling_configs": 2,
92 "rocm.ck_dir": self.ck_dir,
99 @unittest.skipIf(not torch.version.hip, "ROCM only")
120 assert "rocm" in dir(config)
128 "rocm.n_max_profiling_configs": 2,
129 "rocm.ck_dir": self.ck_dir,
146 @unittest.skipIf(not torch.version.hip, "ROCM only")
165 assert "rocm" in dir(config)
[all …]
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | BUILD | 2 # ROCm-platform specific StreamExecutor support code.
11 load("@local_config_rocm//rocm:build_defs.bzl", "if_rocm_is_configured", "rocm_copts")
61 "@local_config_rocm//rocm:rocm_headers",
71 "@local_config_rocm//rocm:rocm_headers",
163 "@local_config_rocm//rocm:rocblas",
175 "@local_config_rocm//rocm:rocm_headers",
211 "@local_config_rocm//rocm:rocm_headers",
219 "@local_config_rocm//rocm:hipfft",
243 "@local_config_rocm//rocm:rocm_headers",
251 "@local_config_rocm//rocm:miopen",
[all …]
|
| D | rocm_activation.h | 16 // This file contains APIs that assume a StreamExecutor is backed by ROCM.
17 // It reaches into the ROCM implementation to activate an underlying ROCM
20 // Having this file separate from rocm/rocm_gpu_executor.h means that dependent
21 // code does not also have to depend on rocm.h.
32 namespace rocm {
36 } // namespace rocm
|
| D | rocm_fft.h | 16 // ROCM-specific support for FFT functionality -- this wraps the rocFFT library
17 // capabilities, and is only included into ROCM implementation code -- it will
18 // not introduce rocm headers into other code.
25 #include "rocm/include/hipfft/hipfft.h"
26 #include "rocm/rocm_config.h"
104 // FFT support for ROCM platform via rocFFT library.
111 // ROCM context is active.
113 // Thread-safe. The ROCM context associated with all operations is the ROCM
|
| /external/pytorch/.github/workflows/ |
| D | rocm.yml | 1 name: rocm
9 - ciflow/rocm/*
39 docker-image-name: pytorch-linux-focal-rocm-n-py3
40 sync-tag: rocm-build
43 { config: "default", shard: 1, num_shards: 6, runner: "linux.rocm.gpu.2" },
44 { config: "default", shard: 2, num_shards: 6, runner: "linux.rocm.gpu.2" },
45 { config: "default", shard: 3, num_shards: 6, runner: "linux.rocm.gpu.2" },
46 { config: "default", shard: 4, num_shards: 6, runner: "linux.rocm.gpu.2" },
47 { config: "default", shard: 5, num_shards: 6, runner: "linux.rocm.gpu.2" },
48 { config: "default", shard: 6, num_shards: 6, runner: "linux.rocm.gpu.2" },
|
| /external/pytorch/.github/scripts/amd/ |
| D | package_triton_wheel.sh | 3 # Set ROCM_HOME isn't available, use ROCM_PATH if set or /opt/rocm
4 ROCM_HOME="${ROCM_HOME:-${ROCM_PATH:-/opt/rocm}}"
6 # Find rocm_version.h header file for ROCm version extract
7 rocm_version_h="${ROCM_HOME}/include/rocm-core/rocm_version.h"
18 # Extract major, minor and patch ROCm version numbers
23 echo "ROCm version: $ROCM_INT"
48 # Required ROCm libraries
55 # Required ROCm libraries - ROCm 6.0
|
| /external/pytorch/docs/source/notes/ |
| D | hip.rst | 3 HIP (ROCm) semantics
6 ROCm\ |trade| is AMD’s open source software platform for GPU-accelerated high
7 performance computing and machine learning. HIP is ROCm's C++ dialect designed
24 cuda0 = torch.device('cuda:0') # 'rocm' or 'hip' are not valid, use 'cuda'
77 TensorFloat-32(TF32) on ROCm
80 TF32 is not supported on ROCm.
90 ``rocm-smi``. You can use :meth:`~torch.cuda.memory_allocated` and
120 Currently, only the "nccl" and "gloo" backends for torch.distributed are supported on ROCm.
135 ``#if defined(CUDA_VERSION) && CUDA_VERSION >= 11000`` to implicitly exclude ROCm/HIP,
137 use the following to not take the code path for ROCm/HIP:
[all …]
|
| /external/pytorch/.ci/docker/ubuntu-rocm/ |
| D | Dockerfile | 63 # Install rocm
73 ENV ROCM_PATH /opt/rocm
74 ENV PATH /opt/rocm/bin:$PATH
75 ENV PATH /opt/rocm/hcc/bin:$PATH
76 ENV PATH /opt/rocm/hip/bin:$PATH
77 ENV PATH /opt/rocm/opencl/bin:$PATH
78 ENV PATH /opt/rocm/llvm/bin:$PATH
79 ENV MAGMA_HOME /opt/rocm/magma
114 RUN ["/bin/bash", "-c", "./install_aotriton.sh /opt/rocm && rm -rf install_aotriton.sh aotriton_ver…
115 ENV AOTRITON_INSTALLED_PREFIX /opt/rocm/aotriton
|
| /external/tensorflow/tensorflow/tools/toolchains/remote_config/ |
| D | containers.bzl | 10 # ROCM, probably not all of them still in use
11 …"rocm-ubuntu18.04-manylinux2010-multipython": "sha256:6e953a09b145df338bcb03e9e36f99b291140c29b72d…
12 …"rocm-ubuntu20.04-manylinux2014-multipython": "sha256:26720ebae4d6d12b1fca529616bfacfd0460990d4725…
101 …# Built with //tensorflow/tools/ci_build/Dockerfile.rbe.rocm-ubuntu18.04-manylinux2010-multipython.
102 "rocm-ubuntu18.04-manylinux2010-multipython": {
104 "repository": "tensorflow-testing/nosla-rocm-ubuntu18.04-manylinux2010-multipython",
105 "digest": container_digests["rocm-ubuntu18.04-manylinux2010-multipython"],
108 …# Built with //tensorflow/tools/ci_build/Dockerfile.rbe.rocm-ubuntu20.04-manylinux2014-multipython.
109 "rocm-ubuntu20.04-manylinux2014-multipython": {
111 "repository": "tensorflow-testing/nosla-rocm-ubuntu20.04-manylinux2014-multipython",
[all …]
|
| /external/pytorch/.ci/docker/centos-rocm/ |
| D | Dockerfile | 72 # Install rocm
83 ENV PATH /opt/rocm/bin:$PATH
84 ENV PATH /opt/rocm/hcc/bin:$PATH
85 ENV PATH /opt/rocm/hip/bin:$PATH
86 ENV PATH /opt/rocm/opencl/bin:$PATH
87 ENV PATH /opt/rocm/llvm/bin:$PATH
88 ENV MAGMA_HOME /opt/rocm/magma
120 RUN ["/bin/bash", "-c", "./install_aotriton.sh /opt/rocm && rm -rf install_aotriton.sh aotriton_ver…
121 ENV AOTRITON_INSTALLED_PREFIX /opt/rocm/aotriton
|
| /external/tensorflow/tensorflow/tsl/platform/default/build_config/ |
| D | BUILD | 48 "//tensorflow/stream_executor/rocm:rocm_activation",
59 "//tensorflow/stream_executor/rocm:rocm_platform_id",
66 "//tensorflow/stream_executor/rocm:all_runtime",
92 ":rocm",
94 ["//tensorflow/stream_executor/rocm:all_runtime"],
112 "//tensorflow/stream_executor/rocm:rocm_platform_id",
225 name = "rocm",
229 "-Wl,-rpath,../local_config_rocm/rocm/rocm/lib",
|
| /external/pytorch/.github/actions/setup-rocm/ |
| D | action.yml | 1 name: Setup ROCm host
3 description: Set up ROCm host for CI
37 cat /etc/apt/sources.list.d/rocm.list || true
38 cat /opt/rocm/.info/version || true
41 - name: Runner health check rocm-smi
45 rocm-smi
86 - name: ROCm set GPU_FLAG
|
| /external/pytorch/.github/templates/ |
| D | linux_binary_build_workflow.yml.j2 | 84 {%- if config["gpu_arch_type"] != "rocm" %}
95 {%- elif config["gpu_arch_type"] == "rocm" %}
96 runs_on: linux.rocm.gpu
105 runs-on: linux.rocm.gpu
109 - name: Setup ROCm
110 uses: ./.github/actions/setup-rocm
118 - name: ROCm set GPU_FLAG
127 - name: Teardown ROCm
128 uses: ./.github/actions/teardown-rocm
|