Skip to content

GitLab

Commit d1167d95 authored by rusty1s's avatar rusty1s
Browse files

typo

parent 9f159a77
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 1 deletion
torch_scatter/kernel/common.cuh
View file @ d1167d95
......@@ -24,7 +24,6 @@ struct TensorInfo {
#define KERNEL_LOOP(I, N) \
for (int I = blockIdx.x * blockDim.x + threadIdx.x; I < N; i += blockDim.x * gridDim.x)
/* #define KERNEL_RUN(NAME, DIMS, N, PARAMS) \ */
#define KERNEL_RUN(NAME, DIMS, N, ...) { \
int grid = GET_BLOCKS(N); \
cudaStream_t stream = THCState_getCurrentStream(state); \
......@@ -44,4 +43,6 @@ static inline __device__ bool eq(int32_t a, int32_t b) { return a == b; }
static inline __device__ bool eq(int64_t a, int64_t b) { return a == b; }
static inline __device__ bool eq( float a, float b) { return a == b; }
static inline __device__ bool eq( double a, double b) { return a == b; }
#ifdef CUDA_HALF_TENSOR
static inline __device__ bool eq(half a, half b) { return __half2float(a) == __half2float(b); }
#endif
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment