File tree Expand file tree Collapse file tree 2 files changed +2
-2
lines changed
Expand file tree Collapse file tree 2 files changed +2
-2
lines changed Original file line number Diff line number Diff line change 11FROM rocm/pytorch:rocm6.4.4_ubuntu24.04_py3.12_pytorch_release_2.7.1
22ARG ROCM_PATH=/opt/rocm
33ARG MIGRAPHX_BRANCH=master
4- ARG GPU_ARCH="gfx900;gfx906; gfx908;gfx90a;gfx1030;gfx1100;gfx1101;gfx1102;gfx940;gfx941; gfx942;gfx950"
4+ ARG GPU_ARCH="gfx908;gfx90a;gfx1030;gfx1100;gfx1101;gfx1102;gfx942;gfx950;gfx1201 "
55
66COPY . /workspace/torch_migraphx
77
Original file line number Diff line number Diff line change @@ -2,7 +2,7 @@ FROM rocm/pytorch:rocm6.4.4_ubuntu24.04_py3.12_pytorch_release_2.7.1
22
33ARG ROCM_PATH=/opt/rocm
44ARG MIGRAPHX_BRANCH=develop
5- ARG GPU_ARCH="gfx900;gfx906; gfx908;gfx90a;gfx1030;gfx1100;gfx1101;gfx1102;gfx940;gfx941; gfx942"
5+ ARG GPU_ARCH="gfx908;gfx90a;gfx1030;gfx1100;gfx1101;gfx1102;gfx950; gfx942;gfx1201 "
66
77# Install Dependencies: MIGraphX
88# Install rbuild
You can’t perform that action at this time.
0 commit comments