ExtensionFunctionsTesting.hpp

Go to the documentation of this file.

/*

  Compile-time UDFs. Used in RBC: rbc/tests/test_omnisci_device_selection.py

 */

#define CPU_DEVICE_CODE 0x637075  // 'cpu' in hex
#define GPU_DEVICE_CODE 0x677075  // 'gpu' in hex

EXTENSION_NOINLINE
int32_t ct_device_selection_udf_any(int32_t input) {
#ifdef __CUDACC__
  return GPU_DEVICE_CODE;
#else
  return CPU_DEVICE_CODE;
#endif
}

#ifndef __CUDACC__

EXTENSION_NOINLINE_HOST
int32_t ct_device_selection_udf_cpu__cpu_(int32_t input) {
  return CPU_DEVICE_CODE;
}

#endif

EXTENSION_NOINLINE
int32_t ct_device_selection_udf_gpu__gpu_(int32_t input) {
  return GPU_DEVICE_CODE;
}

#ifndef __CUDACC__

EXTENSION_NOINLINE_HOST
int32_t ct_device_selection_udf_both__cpu_(int32_t input) {
  return CPU_DEVICE_CODE;
}

#endif

EXTENSION_NOINLINE
int32_t ct_device_selection_udf_both__gpu_(int32_t input) {
  return GPU_DEVICE_CODE;
}

#ifndef __CUDACC__

#include <chrono>
#include <thread>
EXTENSION_NOINLINE
int32_t ct_sleep_us__cpu_(int64_t usec) {
  std::this_thread::sleep_for(std::chrono::microseconds(usec));
  return usec;
}

#endif  // #ifndef __CUDACC__

#undef CPU_DEVICE_CODE
#undef GPU_DEVICE_CODE