HIP: fix GPU_TARGETS (#16642)
This commit is contained in:
parent
e56abd2098
commit
ee09828cb0
|
|
@ -75,7 +75,7 @@ if [ ! -z ${GG_BUILD_ROCM} ]; then
|
||||||
exit 1
|
exit 1
|
||||||
fi
|
fi
|
||||||
|
|
||||||
CMAKE_EXTRA="${CMAKE_EXTRA} -DAMDGPU_TARGETS=${GG_BUILD_AMDGPU_TARGETS}"
|
CMAKE_EXTRA="${CMAKE_EXTRA} -DGPU_TARGETS=${GG_BUILD_AMDGPU_TARGETS}"
|
||||||
fi
|
fi
|
||||||
|
|
||||||
if [ ! -z ${GG_BUILD_SYCL} ]; then
|
if [ ! -z ${GG_BUILD_SYCL} ]; then
|
||||||
|
|
|
||||||
|
|
@ -28,8 +28,10 @@ if (CXX_IS_HIPCC)
|
||||||
" Prefer setting the HIP compiler directly. See README for details.")
|
" Prefer setting the HIP compiler directly. See README for details.")
|
||||||
endif()
|
endif()
|
||||||
else()
|
else()
|
||||||
# Forward AMDGPU_TARGETS to CMAKE_HIP_ARCHITECTURES.
|
# Forward (AMD)GPU_TARGETS to CMAKE_HIP_ARCHITECTURES.
|
||||||
if (AMDGPU_TARGETS AND NOT CMAKE_HIP_ARCHITECTURES)
|
if(GPU_TARGETS AND NOT CMAKE_HIP_ARCHITECTURES)
|
||||||
|
set(CMAKE_HIP_ARCHITECTURES ${GPU_TARGETS})
|
||||||
|
elseif(AMDGPU_TARGETS AND NOT CMAKE_HIP_ARCHITECTURES)
|
||||||
set(CMAKE_HIP_ARCHITECTURES ${AMDGPU_TARGETS})
|
set(CMAKE_HIP_ARCHITECTURES ${AMDGPU_TARGETS})
|
||||||
endif()
|
endif()
|
||||||
cmake_minimum_required(VERSION 3.21)
|
cmake_minimum_required(VERSION 3.21)
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue