From 8d88a6ac7f15cfb0f075fb420ec24680872cd60d Mon Sep 17 00:00:00 2001 From: Jeff Daily Date: Wed, 1 Dec 2021 22:28:52 -0800 Subject: [PATCH] add --amdgpu-target=gfx90a (#9820) --- cmake/onnxruntime_providers.cmake | 3 +++ 1 file changed, 3 insertions(+) diff --git a/cmake/onnxruntime_providers.cmake b/cmake/onnxruntime_providers.cmake index a82629bed4..e29dd8f075 100644 --- a/cmake/onnxruntime_providers.cmake +++ b/cmake/onnxruntime_providers.cmake @@ -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})