Merge pull request #11 from XSShawnZeng/Tegra_Samples_Cmake_Transition

Update separate SM list for cdp samples and update the watershedSegmentationNPP for build failed of Bug 4668487
This commit is contained in:
Rob Armstrong 2025-01-21 09:38:41 -08:00 committed by GitHub
commit 3de3f2f46c
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
6 changed files with 31 additions and 6 deletions

View File

@ -9,7 +9,12 @@ find_package(CUDAToolkit REQUIRED)
set(CMAKE_POSITION_INDEPENDENT_CODE ON)
if(CMAKE_SYSTEM_PROCESSOR STREQUAL "aarch64")
set(CMAKE_CUDA_ARCHITECTURES 61 70 75 80 86 87 90)
# Need to differentiate Tegra_aarch64 and other aarch64 systems(sbsa_aarch64) as they have different CUDA_ARCHITECTURES list
if(${BUILD_TEGRA})
set(CMAKE_CUDA_ARCHITECTURES 72 87 101)
else()
set(CMAKE_CUDA_ARCHITECTURES 61 70 75 80 86 90)
endif()
else()
set(CMAKE_CUDA_ARCHITECTURES 50 52 60 61 70 75 80 86 89 90)
endif()

View File

@ -9,7 +9,12 @@ find_package(CUDAToolkit REQUIRED)
set(CMAKE_POSITION_INDEPENDENT_CODE ON)
if(CMAKE_SYSTEM_PROCESSOR STREQUAL "aarch64")
set(CMAKE_CUDA_ARCHITECTURES 61 70 75 80 86 87 90)
# Need to differentiate Tegra_aarch64 and other aarch64 systems(sbsa_aarch64) as they have different CUDA_ARCHITECTURES list
if(${BUILD_TEGRA})
set(CMAKE_CUDA_ARCHITECTURES 72 87 101)
else()
set(CMAKE_CUDA_ARCHITECTURES 61 70 75 80 86 90)
endif()
else()
set(CMAKE_CUDA_ARCHITECTURES 50 52 60 61 70 75 80 86 89 90)
endif()

View File

@ -9,7 +9,12 @@ find_package(CUDAToolkit REQUIRED)
set(CMAKE_POSITION_INDEPENDENT_CODE ON)
if(CMAKE_SYSTEM_PROCESSOR STREQUAL "aarch64")
set(CMAKE_CUDA_ARCHITECTURES 61 70 75 80 86 87 90)
# Need to differentiate Tegra_aarch64 and other aarch64 systems(sbsa_aarch64) as they have different CUDA_ARCHITECTURES list
if(${BUILD_TEGRA})
set(CMAKE_CUDA_ARCHITECTURES 72 87 101)
else()
set(CMAKE_CUDA_ARCHITECTURES 61 70 75 80 86 90)
endif()
else()
set(CMAKE_CUDA_ARCHITECTURES 50 52 60 61 70 75 80 86 89 90)
endif()

View File

@ -9,7 +9,12 @@ find_package(CUDAToolkit REQUIRED)
set(CMAKE_POSITION_INDEPENDENT_CODE ON)
if(CMAKE_SYSTEM_PROCESSOR STREQUAL "aarch64")
set(CMAKE_CUDA_ARCHITECTURES 61 70 75 80 86 87 90)
# Need to differentiate Tegra_aarch64 and other aarch64 systems(sbsa_aarch64) as they have different CUDA_ARCHITECTURES list
if(${BUILD_TEGRA})
set(CMAKE_CUDA_ARCHITECTURES 72 87 101)
else()
set(CMAKE_CUDA_ARCHITECTURES 61 70 75 80 86 90)
endif()
else()
set(CMAKE_CUDA_ARCHITECTURES 50 52 60 61 70 75 80 86 89 90)
endif()

View File

@ -9,7 +9,12 @@ find_package(CUDAToolkit REQUIRED)
set(CMAKE_POSITION_INDEPENDENT_CODE ON)
if(CMAKE_SYSTEM_PROCESSOR STREQUAL "aarch64")
set(CMAKE_CUDA_ARCHITECTURES 61 70 75 80 86 87 90)
# Need to differentiate Tegra_aarch64 and other aarch64 systems(sbsa_aarch64) as they have different CUDA_ARCHITECTURES list
if(${BUILD_TEGRA})
set(CMAKE_CUDA_ARCHITECTURES 72 87 101)
else()
set(CMAKE_CUDA_ARCHITECTURES 61 70 75 80 86 90)
endif()
else()
set(CMAKE_CUDA_ARCHITECTURES 50 52 60 61 70 75 80 86 89 90)
endif()

View File

@ -173,7 +173,7 @@ int
main( int argc, char** argv )
{
int aSegmentationScratchBufferSize[NUMBER_OF_IMAGES];
size_t aSegmentationScratchBufferSize[NUMBER_OF_IMAGES];
int aSegmentLabelsOutputBufferSize[NUMBER_OF_IMAGES];
cudaError_t cudaError;