From ca9451d89169a0de54be578db8b08d7e8a32c1e7 Mon Sep 17 00:00:00 2001 From: Lakshmi Kumar Date: Tue, 25 Feb 2025 11:52:17 -0800 Subject: [PATCH] Check ROCm targets (#513) * rocm version check for targets * check rocm targets for gfx [ROCm/rocdecode commit: ad0bec2f145f5fb18da3ad3d4b995aa96e9ba29e] --- .../samples/videoDecodeRGB/CMakeLists.txt | 16 +++++++++++++++- 1 file changed, 15 insertions(+), 1 deletion(-) diff --git a/projects/rocdecode/samples/videoDecodeRGB/CMakeLists.txt b/projects/rocdecode/samples/videoDecodeRGB/CMakeLists.txt index 65060d6e70..4f3fd9e17e 100644 --- a/projects/rocdecode/samples/videoDecodeRGB/CMakeLists.txt +++ b/projects/rocdecode/samples/videoDecodeRGB/CMakeLists.txt @@ -61,7 +61,21 @@ else() endif() # Set supported GPU Targets -set(DEFAULT_GPU_TARGETS "gfx908;gfx90a;gfx942;gfx950;gfx1030;gfx1031;gfx1032;gfx1100;gfx1101;gfx1102;gfx1200;gfx1201") +if(NOT GPU_TARGETS AND NOT AMDGPU_TARGETS) + find_package(ROCmCMakeBuildTools QUIET) + if(NOT ROCmCMakeBuildTools_FOUND) + find_package(ROCM QUIET) + endif() + include(ROCMCheckTargetIds OPTIONAL RESULT_VARIABLE HAS_ROCM_CHECK_TARGET_IDS) + + set(OPTIONAL_GPU_TARGETS "gfx950;gfx1200;gfx1201") + if(HAS_ROCM_CHECK_TARGET_IDS) + rocm_check_target_ids(OPTIONAL_GPU_TARGETS_AVAILABLE TARGETS ${OPTIONAL_GPU_TARGETS}) + else() # if we don't have rocm_check_target_ids, just assume the targets are available + set(OPTIONAL_GPU_TARGETS_AVAILABLE "${OPTIONAL_GPU_TARGETS}") + endif() + set(DEFAULT_GPU_TARGETS "gfx908;gfx90a;gfx942;gfx1030;gfx1031;gfx1032;gfx1100;gfx1101;gfx1102;${OPTIONAL_GPU_TARGETS_AVAILABLE}") +endif() # Set AMD GPU_TARGETS if((AMDGPU_TARGETS OR DEFINED ENV{AMDGPU_TARGETS}) AND (NOT GPU_TARGETS))