Skip to content

GitLab

Commit 7af61dbe authored by rusty1s's avatar rusty1s
Browse files

typo

parent 9fcc4d8a
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 5 deletions
csrc/cuda/utils.cuh
View file @ 7af61dbe
...@@ -6,13 +6,10 @@ ...@@ -6,13 +6,10 @@
AT_ASSERTM(x.device().is_cuda(), #x " must be CUDA tensor") AT_ASSERTM(x.device().is_cuda(), #x " must be CUDA tensor")
#define CHECK_INPUT(x) AT_ASSERTM(x, "Input mismatch") #define CHECK_INPUT(x) AT_ASSERTM(x, "Input mismatch")
#ifndef _WIN32 __device__ __inline__ at::Half
__device__ __inline__ at::Half __shfl_sync(const unsigned mask, __shfl_sync(const unsigned mask, const at::Half var, const int srcLane) {
const at::Half var,
const unsigned int srcLane) {
return __shfl_sync(mask, (__half)var, srcLane); return __shfl_sync(mask, (__half)var, srcLane);
} }
#endif
__device__ __inline__ at::Half __shfl_down_sync(const unsigned mask, __device__ __inline__ at::Half __shfl_down_sync(const unsigned mask,
const at::Half var, const at::Half var,
......
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