fix parsing code name for MI200/300

This commit is contained in:
ghostplant 2023-09-05 15:29:49 +08:00
Родитель 9fc5dcec3f
Коммит 0a6df04423
2 изменённых файлов: 2 добавлений и 2 удалений

Просмотреть файл

@ -39,7 +39,7 @@ def do_native_translation_v2(codeset, **kwargs):
amdgfx = os.environ['AMDGFX']
else:
amdgfx = kwargs['attrs'].device_props.compute_version.split('.')
if int(amdgfx[0]) < 10:
if int(amdgfx[0]) < 10 and int(amdgfx[1]) == 10:
amdgfx = 'gfx%u%02x' % (int(amdgfx[0]), int(amdgfx[1]))
else:
amdgfx = 'gfx%u%02u' % (int(amdgfx[0]), int(amdgfx[1]))

Просмотреть файл

@ -21,7 +21,7 @@ RUN bash -c 'rm -rf ~/.local/antares/3rdparty/tvm/build/{CMake*,Makefile,cmake_i
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,.??*}'
RUN echo '' > ~/.local/antares/3rdparty/tvm/python/tvm/relay/__init__.py
ENV ANTARES_VERSION 0.3.23.1
ENV ANTARES_VERSION 0.3.23.2
RUN cd ~ && git clone https://github.com/microsoft/antares --branch latest --single-branch --depth 1 antares_core && mv ~/.local/antares/3rdparty antares_core
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