Skip to content

GitLab

Commit a75b2bad authored by yanyan's avatar yanyan
Browse files

fix __hadd2 is undefined

parent 5fb1e359
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
include/spconv/reordering.cu.h
View file @ a75b2bad
...@@ -17,6 +17,7 @@ ...@@ -17,6 +17,7 @@
#include <THC/THCAtomics.cuh> #include <THC/THCAtomics.cuh>
#include <THC/THCNumerics.cuh> #include <THC/THCNumerics.cuh>
#include <tensorview/kernel_utils.h> #include <tensorview/kernel_utils.h>
#include <cuda_fp16.h>
#if PYTORCH_VERSION < 10500 #if PYTORCH_VERSION < 10500
#define TH_ATOMIC_ADD atomicAdd #define TH_ATOMIC_ADD atomicAdd
......
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