fix compilation bug
This commit is contained in:
parent
386bfe5566
commit
00ea219123
@ -2,6 +2,7 @@
|
||||
#include <cuda_runtime.h>
|
||||
#include <torch/extension.h>
|
||||
|
||||
#include <ATen/cuda/CUDAContext.h>
|
||||
#include <THC/THCAtomics.cuh>
|
||||
#include <vector>
|
||||
|
||||
|
||||
@ -2,6 +2,7 @@
|
||||
#include <cuda_runtime.h>
|
||||
#include <torch/extension.h>
|
||||
|
||||
#include <ATen/cuda/CUDAContext.h>
|
||||
#include <THC/THCAtomics.cuh>
|
||||
#include <vector>
|
||||
|
||||
|
||||
@ -2,6 +2,7 @@
|
||||
#include <cuda_runtime.h>
|
||||
#include <torch/extension.h>
|
||||
|
||||
#include <ATen/cuda/CUDAContext.h>
|
||||
#include <THC/THCAtomics.cuh>
|
||||
#include <vector>
|
||||
|
||||
|
||||
Loading…
Reference in New Issue
Block a user