fix build

khiemauto · Jan 20, 2020 · a3bebeb · a3bebeb
1 parent 6d012ee
commit a3bebeb
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/cuda/dcn_v2_cuda.cu b/src/cuda/dcn_v2_cuda.cu
@@ -8,7 +8,7 @@
 #include <THC/THCAtomics.cuh>
 #include <THC/THCDeviceUtils.cuh>
 
-extern THCState *state;
+THCState *state = at::globalContext().lazyInitCUDA();
 
 // author: Charles Shang
 // https://github.com/torch/cunn/blob/master/lib/THCUNN/generic/SpatialConvolutionMM.cu