Global.h

Go to the documentation of this file.

 
#pragma once
 
#include <torch/extension.h>
 
#ifdef __CUDACC__
#include <cuda.h>
#include <cuda_runtime.h>
#else
#define __host__
 
#define __device__
#endif
 
namespace reg23 {
 
template <typename T> __host__ __device__ T Square(const T &x) { return x * x; }
 
template <typename T> __host__ __device__ T Modulo(const T &x, const T &y) { return (x % y + y) % y; }
 
 
template <typename T> __host__ __device__ T Sign(const T &x) { return static_cast<T>((x > 0) - (x < 0)); }
 
} // namespace reg23