Skip to content

Commit 9a88d5f

Browse files
authored
Merge pull request #1432 from kiritigowda/kg/gpu-targets
GPU Targets - Updates
2 parents fd6ea1a + 903af80 commit 9a88d5f

File tree

1 file changed

+18
-9
lines changed

1 file changed

+18
-9
lines changed

CMakeLists.txt

Lines changed: 18 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -136,20 +136,29 @@ endif()
136136
if("${BACKEND}" STREQUAL "HIP")
137137
set(LOOM OFF)
138138
# Set supported GPU Targets
139-
set(DEFAULT_AMDGPU_TARGETS "gfx908;gfx90a;gfx940;gfx941;gfx942;gfx1030;gfx1031;gfx1032;gfx1100;gfx1101;gfx1102")
139+
set(DEFAULT_GPU_TARGETS "gfx908;gfx90a;gfx940;gfx941;gfx942;gfx1030;gfx1031;gfx1032;gfx1100;gfx1101;gfx1102")
140140
if (BUILD_WITH_AMD_ADVANCE)
141-
set(DEFAULT_AMDGPU_TARGETS ${DEFAULT_AMDGPU_TARGETS} "gfx1200;gfx1201")
141+
set(DEFAULT_GPU_TARGETS ${DEFAULT_GPU_TARGETS} "gfx1200;gfx1201")
142142
endif()
143143

144-
# Set AMDGPU_TARGETS
145-
if(DEFINED ENV{AMDGPU_TARGETS})
146-
set(AMDGPU_TARGETS $ENV{AMDGPU_TARGETS} CACHE STRING "List of specific machine types for library to target")
147-
elseif(AMDGPU_TARGETS)
148-
message("-- ${White}${PROJECT_NAME} -- AMDGPU_TARGETS set with -D option${ColourReset}")
144+
# Set AMD GPU_TARGETS
145+
if((AMDGPU_TARGETS OR DEFINED ENV{AMDGPU_TARGETS}) AND (NOT GPU_TARGETS))
146+
message("-- ${Red}${PROJECT_NAME} DEPRECATION -- AMDGPU_TARGETS use is deprecated. Use GPU_TARGETS${ColourReset}")
147+
if(DEFINED ENV{AMDGPU_TARGETS})
148+
set(GPU_TARGETS $ENV{AMDGPU_TARGETS} CACHE STRING "List of specific machine types for library to target")
149+
else()
150+
set(GPU_TARGETS ${AMDGPU_TARGETS})
151+
endif()
152+
endif()
153+
154+
if(DEFINED ENV{GPU_ARCHS})
155+
set(GPU_TARGETS $ENV{GPU_ARCHS} CACHE STRING "List of specific machine types for library to target")
156+
elseif(GPU_TARGETS)
157+
message("-- ${White}${PROJECT_NAME} -- GPU_TARGETS set with -D option${ColourReset}")
149158
else()
150-
set(AMDGPU_TARGETS "${DEFAULT_AMDGPU_TARGETS}" CACHE STRING "List of specific machine types for library to target")
159+
set(GPU_TARGETS "${DEFAULT_GPU_TARGETS}" CACHE STRING "List of specific machine types for library to target")
151160
endif()
152-
message("-- ${White}${PROJECT_NAME} -- AMDGPU_TARGETS: ${AMDGPU_TARGETS}${ColourReset}")
161+
message("-- ${White}${PROJECT_NAME} -- AMD GPU_TARGETS: ${GPU_TARGETS}${ColourReset}")
153162
endif()
154163

155164
message("-- ${Cyan}MIVisionX Developer Options${ColourReset}")

0 commit comments

Comments
 (0)