"vscode:/vscode.git/clone" did not exist on "ee71a3b63bb7dd9c4a7ba52206966727510e41b8"
Commit 8873a510 authored by fengzch-das's avatar fengzch-das
Browse files

fix: getCurrentStream() not in at::cuda

parent 44fd61fa
...@@ -61,6 +61,7 @@ ...@@ -61,6 +61,7 @@
// modified from https://github.com/chengdazhi/Deformable-Convolution-V2-PyTorch/blob/mmdetection/mmdet/ops/dcn/src/deform_conv_cuda_kernel.cu // modified from https://github.com/chengdazhi/Deformable-Convolution-V2-PyTorch/blob/mmdetection/mmdet/ops/dcn/src/deform_conv_cuda_kernel.cu
#include <ATen/ATen.h> #include <ATen/ATen.h>
#include <c10/cuda/CUDAStream.h>
#include <ATen/cuda/CUDAContext.h> #include <ATen/cuda/CUDAContext.h>
#include <THC/THCAtomics.cuh> #include <THC/THCAtomics.cuh>
#include <stdio.h> #include <stdio.h>
......
...@@ -62,6 +62,7 @@ ...@@ -62,6 +62,7 @@
#include <ATen/ATen.h> #include <ATen/ATen.h>
#include <ATen/cuda/CUDAContext.h> #include <ATen/cuda/CUDAContext.h>
#include <c10/cuda/CUDAStream.h>
#include <THC/THCAtomics.cuh> #include <THC/THCAtomics.cuh>
#include <stdio.h> #include <stdio.h>
#include <math.h> #include <math.h>
......
...@@ -9,6 +9,7 @@ ...@@ -9,6 +9,7 @@
// modify from https://github.com/chengdazhi/Deformable-Convolution-V2-PyTorch/blob/mmdetection/mmdet/ops/dcn/src/cuda/deform_psroi_pooling_cuda.cu // modify from https://github.com/chengdazhi/Deformable-Convolution-V2-PyTorch/blob/mmdetection/mmdet/ops/dcn/src/cuda/deform_psroi_pooling_cuda.cu
#include <ATen/ATen.h> #include <ATen/ATen.h>
#include <c10/cuda/CUDAStream.h>
#include <THC/THCAtomics.cuh> #include <THC/THCAtomics.cuh>
#include <stdio.h> #include <stdio.h>
#include <math.h> #include <math.h>
......
...@@ -9,6 +9,7 @@ ...@@ -9,6 +9,7 @@
// modify from https://github.com/chengdazhi/Deformable-Convolution-V2-PyTorch/blob/mmdetection/mmdet/ops/dcn/src/cuda/deform_psroi_pooling_cuda.cu // modify from https://github.com/chengdazhi/Deformable-Convolution-V2-PyTorch/blob/mmdetection/mmdet/ops/dcn/src/cuda/deform_psroi_pooling_cuda.cu
#include <ATen/ATen.h> #include <ATen/ATen.h>
#include <c10/cuda/CUDAStream.h>
#include <THC/THCAtomics.cuh> #include <THC/THCAtomics.cuh>
#include <stdio.h> #include <stdio.h>
#include <math.h> #include <math.h>
......
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