зеркало из https://github.com/microsoft/antares.git
fix parsing code name for MI200/300
This commit is contained in:
Родитель
9fc5dcec3f
Коммит
0a6df04423
|
@ -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
|
||||
|
|
Загрузка…
Ссылка в новой задаче