Revert "Complication issue with pytorch 1.1"

This reverts commit f16357c.
hzli-ucas · Jan 25, 2019 · 596bcc6 · 596bcc6
1 parent a8c9621
commit 596bcc6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pytorch_binding/src/binding.cpp b/pytorch_binding/src/binding.cpp
@@ -7,7 +7,7 @@
 
 #ifdef WARPCTC_ENABLE_GPU
 	#include "ATen/cuda/CUDAContext.h"
-	#include <c10/cuda/CUDAGuard.h>
+	#include "ATen/cuda/CUDAGuard.h"
 	#include "ATen/cuda/CUDAEvent.h"
 
     #include "THC.h"