Skip to content

Commit 0a6df04

Browse files
committed
fix parsing code name for MI200/300
1 parent 9fc5dce commit 0a6df04

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

backends/c-rocm/config.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -39,7 +39,7 @@ def get_extent(key, defval=1):
3939
amdgfx = os.environ['AMDGFX']
4040
else:
4141
amdgfx = kwargs['attrs'].device_props.compute_version.split('.')
42-
if int(amdgfx[0]) < 10:
42+
if int(amdgfx[0]) < 10 and int(amdgfx[1]) == 10:
4343
amdgfx = 'gfx%u%02x' % (int(amdgfx[0]), int(amdgfx[1]))
4444
else:
4545
amdgfx = 'gfx%u%02u' % (int(amdgfx[0]), int(amdgfx[1]))

docker/Dockerfile.c-base

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -21,7 +21,7 @@ RUN bash -c 'rm -rf ~/.local/antares/3rdparty/tvm/build/{CMake*,Makefile,cmake_i
2121
RUN bash -c 'rm -rf ~/.local/antares/3rdparty/tvm/{src,include,golang,tests,3rdparty,device-stub,apps,conda,docker,docs,gallery,jvm,nnvm,rust,vta,web,cmake,.??*}'
2222
RUN echo '' > ~/.local/antares/3rdparty/tvm/python/tvm/relay/__init__.py
2323

24-
ENV ANTARES_VERSION 0.3.23.1
24+
ENV ANTARES_VERSION 0.3.23.2
2525

2626
RUN cd ~ && git clone https://github.com/microsoft/antares --branch latest --single-branch --depth 1 antares_core && mv ~/.local/antares/3rdparty antares_core
2727
RUN cd ~ && sed -i "s/@VERSION@/${ANTARES_VERSION}/g" /antares/engine/dist-info/METADATA && cp -r /antares/engine/dist-info ~/antares-${ANTARES_VERSION}.dist-info

0 commit comments

Comments
 (0)