fix __hadd2 is undefined

a75b2bad · yanyan · 5fb1e359 · a75b2bad
Commit a75b2bad authored May 29, 2020 by yanyan
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 0 deletions

include/spconv/reordering.cu.h include/spconv/reordering.cu.h +1 -0

No files found.
--- a/include/spconv/reordering.cu.h
+++ b/include/spconv/reordering.cu.h
@@ -17,6 +17,7 @@
 #include <THC/THCAtomics.cuh>
 #include <THC/THCNumerics.cuh>
 #include <tensorview/kernel_utils.h>
+#include <cuda_fp16.h>

 #if PYTORCH_VERSION < 10500
 #define TH_ATOMIC_ADD atomicAdd