Skip to content

GitLab

"cacheflow/vscode:/vscode.git/clone" did not exist on "c9d5b6d4a8b3f51ff6c9eee7eb52bb5149d89b6a"
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 @@
AT_ASSERTM(x.device().is_cuda(), #x " must be CUDA tensor")
#define CHECK_INPUT(x) AT_ASSERTM(x, "Input mismatch")
#ifndef _WIN32
__device__ __inline__ at::Half __shfl_sync(const unsigned mask,
const at::Half var,
const unsigned int srcLane) {
__device__ __inline__ at::Half
__shfl_sync(const unsigned mask, const at::Half var, const int srcLane) {
return __shfl_sync(mask, (__half)var, srcLane);
}
#endif
__device__ __inline__ at::Half __shfl_down_sync(const unsigned mask,
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