Unverified Commit 7a6c9ab4 authored by Alessandro Pacchioni's avatar Alessandro Pacchioni Committed by GitHub
Browse files

used explicit at::Half to __half conversion operator to fix ambiguous conversion error (#220)


Co-authored-by: default avatarAlessandro Pacchioni <apacchioni@ambarella.com>
parent 122811a6
......@@ -8,11 +8,11 @@
__device__ __inline__ at::Half
__shfl_sync(const unsigned mask, const at::Half var, const int srcLane) {
return __shfl_sync(mask, (__half)var, srcLane);
return __shfl_sync(mask, var.operator __half(), srcLane);
}
__device__ __inline__ at::Half __shfl_down_sync(const unsigned mask,
const at::Half var,
const unsigned int delta) {
return __shfl_down_sync(mask, (__half)var, delta);
return __shfl_down_sync(mask, var.operator __half(), delta);
}
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