35 lines
1.1 KiB
Plaintext
35 lines
1.1 KiB
Plaintext
|
// SPDX-FileCopyrightText: Copyright (c) 2026, NVIDIA CORPORATION. All rights reserved.
|
||
|
|
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
|
||
|
|
|
||
|
|
// This is a compile-only test that checks whether cub can handle a future (unknown) architecture. Currently, works only
|
||
|
|
// with nvcc.
|
||
|
|
|
||
|
|
#if defined(__NVCC__) && __CUDACC_VER_MAJOR__ >= 13
|
||
|
|
# define TEST_CUDA_ARCH 9900
|
||
|
|
|
||
|
|
// Replace __CUDA_ARCH__ only in device code.
|
||
|
|
# if defined(__CUDA_ARCH__)
|
||
|
|
# undef __CUDA_ARCH__
|
||
|
|
# define __CUDA_ARCH__ TEST_CUDA_ARCH
|
||
|
|
# endif // __CUDA_ARCH__
|
||
|
|
|
||
|
|
// __CUDA_ARCH_LIST__ is defined in both host and device code.
|
||
|
|
# undef __CUDA_ARCH_LIST__
|
||
|
|
# define __CUDA_ARCH_LIST__ TEST_CUDA_ARCH
|
||
|
|
|
||
|
|
// We need to undefine the include guard before including <cuda_runtime.h>. It was defined by us in CMake.
|
||
|
|
# undef __CUDA_RUNTIME_H__
|
||
|
|
# include <cuda_runtime.h>
|
||
|
|
#endif // __NVCC__ && is at least 13.0
|
||
|
|
|
||
|
|
#include <cub/device/device_reduce.cuh>
|
||
|
|
|
||
|
|
#include <cuda/std/cstddef>
|
||
|
|
|
||
|
|
cudaError_t compile_only_fn(void* tmp_storage, cuda::std::size_t tmp_storage_size, int* in, int* out, int nitems)
|
||
|
|
{
|
||
|
|
return cub::DeviceReduce::Sum(tmp_storage, tmp_storage_size, in, out, nitems);
|
||
|
|
}
|
||
|
|
|
||
|
|
int main() {}
|