saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-07-23 19:32:23 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

add --amdgpu-target=gfx90a (#9820)

Browse source
This commit is contained in:
Jeff Daily 2021-12-01 22:28:52 -08:00 committed by GitHub
parent 00c979db4d
commit 8d88a6ac7f
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
cmake/onnxruntime_providers.cmake
View file

@ -1106,6 +1106,9 @@ if (onnxruntime_USE_ROCM)
# Generate GPU code for GFX9 Generation
list(APPEND HIP_CLANG_FLAGS --amdgpu-target=gfx906 --amdgpu-target=gfx908)
if (ROCM_VERSION_DEV_INT GREATER_EQUAL 50000)
list(APPEND HIP_CLANG_FLAGS --amdgpu-target=gfx90a)
endif()
#onnxruntime_add_shared_library_module(onnxruntime_providers_rocm ${onnxruntime_providers_rocm_src})
hip_add_library(onnxruntime_providers_rocm MODULE ${onnxruntime_providers_rocm_src})