HeterogeneousCore/CUDAUtilities/test/cudastdAlgorithm_t.cu - master - githubmirror/cmssw

HeterogeneousCore/CUDAUtilities/test/cudastdAlgorithm_t.cu

30 строк868 B

Vincenzo Innocente

Add common tools for processing on GPUs

05 мар 2020, 14:47

05 мар 2020, 14:47ce7301f

#include <cassert>
#include <iostream>

#include "HeterogeneousCore/CUDAUtilities/interface/cudastdAlgorithm.h"
#include "HeterogeneousCore/CUDAUtilities/interface/requireDevices.h"
#include "HeterogeneousCore/CUDAUtilities/interface/launch.h"

__global__ void testBinaryFind() {
  int data[] = {1, 1, 2, 3, 3, 3, 3, 4, 4, 4, 5, 5, 6};

  auto lower = cuda_std::lower_bound(data, data + 13, 4);
  auto upper = cuda_std::upper_bound(data, data + 12, 4);

  assert(3 == upper - lower);

  // classic binary search, returning a value only if it is present

  constexpr int data2[] = {1, 2, 4, 6, 9, 10};

  assert(data2 + 2 == cuda_std::binary_find(data2, data2 + 6, 4));
  assert(data2 + 6 == cuda_std::binary_find(data2, data2 + 6, 5));
}

void wrapper() { cms::cuda::launch(testBinaryFind, {32, 64}); }

int main() {
  cms::cudatest::requireDevices();

  wrapper();
}