mirror of
https://github.com/NVIDIA/cuda-samples.git
synced 2025-07-02 13:20:30 +08:00
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:
commit
3de3f2f46c
@ -9,7 +9,12 @@ find_package(CUDAToolkit REQUIRED)
|
|||||||
set(CMAKE_POSITION_INDEPENDENT_CODE ON)
|
set(CMAKE_POSITION_INDEPENDENT_CODE ON)
|
||||||
|
|
||||||
if(CMAKE_SYSTEM_PROCESSOR STREQUAL "aarch64")
|
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()
|
else()
|
||||||
set(CMAKE_CUDA_ARCHITECTURES 50 52 60 61 70 75 80 86 89 90)
|
set(CMAKE_CUDA_ARCHITECTURES 50 52 60 61 70 75 80 86 89 90)
|
||||||
endif()
|
endif()
|
||||||
|
@ -9,7 +9,12 @@ find_package(CUDAToolkit REQUIRED)
|
|||||||
set(CMAKE_POSITION_INDEPENDENT_CODE ON)
|
set(CMAKE_POSITION_INDEPENDENT_CODE ON)
|
||||||
|
|
||||||
if(CMAKE_SYSTEM_PROCESSOR STREQUAL "aarch64")
|
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()
|
else()
|
||||||
set(CMAKE_CUDA_ARCHITECTURES 50 52 60 61 70 75 80 86 89 90)
|
set(CMAKE_CUDA_ARCHITECTURES 50 52 60 61 70 75 80 86 89 90)
|
||||||
endif()
|
endif()
|
||||||
|
@ -9,7 +9,12 @@ find_package(CUDAToolkit REQUIRED)
|
|||||||
set(CMAKE_POSITION_INDEPENDENT_CODE ON)
|
set(CMAKE_POSITION_INDEPENDENT_CODE ON)
|
||||||
|
|
||||||
if(CMAKE_SYSTEM_PROCESSOR STREQUAL "aarch64")
|
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()
|
else()
|
||||||
set(CMAKE_CUDA_ARCHITECTURES 50 52 60 61 70 75 80 86 89 90)
|
set(CMAKE_CUDA_ARCHITECTURES 50 52 60 61 70 75 80 86 89 90)
|
||||||
endif()
|
endif()
|
||||||
|
@ -9,7 +9,12 @@ find_package(CUDAToolkit REQUIRED)
|
|||||||
set(CMAKE_POSITION_INDEPENDENT_CODE ON)
|
set(CMAKE_POSITION_INDEPENDENT_CODE ON)
|
||||||
|
|
||||||
if(CMAKE_SYSTEM_PROCESSOR STREQUAL "aarch64")
|
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()
|
else()
|
||||||
set(CMAKE_CUDA_ARCHITECTURES 50 52 60 61 70 75 80 86 89 90)
|
set(CMAKE_CUDA_ARCHITECTURES 50 52 60 61 70 75 80 86 89 90)
|
||||||
endif()
|
endif()
|
||||||
|
@ -9,7 +9,12 @@ find_package(CUDAToolkit REQUIRED)
|
|||||||
set(CMAKE_POSITION_INDEPENDENT_CODE ON)
|
set(CMAKE_POSITION_INDEPENDENT_CODE ON)
|
||||||
|
|
||||||
if(CMAKE_SYSTEM_PROCESSOR STREQUAL "aarch64")
|
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()
|
else()
|
||||||
set(CMAKE_CUDA_ARCHITECTURES 50 52 60 61 70 75 80 86 89 90)
|
set(CMAKE_CUDA_ARCHITECTURES 50 52 60 61 70 75 80 86 89 90)
|
||||||
endif()
|
endif()
|
||||||
|
@ -173,7 +173,7 @@ int
|
|||||||
main( int argc, char** argv )
|
main( int argc, char** argv )
|
||||||
{
|
{
|
||||||
|
|
||||||
int aSegmentationScratchBufferSize[NUMBER_OF_IMAGES];
|
size_t aSegmentationScratchBufferSize[NUMBER_OF_IMAGES];
|
||||||
int aSegmentLabelsOutputBufferSize[NUMBER_OF_IMAGES];
|
int aSegmentLabelsOutputBufferSize[NUMBER_OF_IMAGES];
|
||||||
|
|
||||||
cudaError_t cudaError;
|
cudaError_t cudaError;
|
||||||
|
Loading…
x
Reference in New Issue
Block a user