18 lines
506 B
Plaintext
18 lines
506 B
Plaintext
|
#include <cub/device/device_for.cuh> // internal include of NVTX
|
||
|
|
|
||
|
|
#include <thrust/iterator/counting_iterator.h>
|
||
|
|
|
||
|
|
#include <cuda/iterator>
|
||
|
|
#include <cuda/std/functional>
|
||
|
|
|
||
|
|
#include <nvtx3/nvtx3.hpp> // user-side include of NVTX, retrieved elsewhere
|
||
|
|
|
||
|
|
int main()
|
||
|
|
{
|
||
|
|
nvtx3::scoped_range range("user-range"); // user-side use of unversioned NVTX API
|
||
|
|
|
||
|
|
cuda::counting_iterator<int> it{0};
|
||
|
|
cub::DeviceFor::ForEach(it, it + 16, ::cuda::std::negate<int>{}); // internal use of NVTX
|
||
|
|
cudaDeviceSynchronize();
|
||
|
|
}
|