Skip to content

Commit 13cdd30

Browse files
authored
navi48-support-missing-in-Dockerfile (#273)
* navi48-support-missing-in-Dockerfile * remove unsupported gpu target * add gfx950
1 parent 7a7ea51 commit 13cdd30

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

Dockerfile

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
FROM rocm/pytorch:rocm6.4.4_ubuntu24.04_py3.12_pytorch_release_2.7.1
22
ARG ROCM_PATH=/opt/rocm
33
ARG 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

66
COPY . /workspace/torch_migraphx
77

docker/dev.Dockerfile

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,7 @@ FROM rocm/pytorch:rocm6.4.4_ubuntu24.04_py3.12_pytorch_release_2.7.1
22

33
ARG ROCM_PATH=/opt/rocm
44
ARG 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

0 commit comments

Comments
 (0)