Deformable batch works

Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com>
2020-04-25 02:47:31 +02:00
parent 13c9dc6620
commit 6e5ab031c4
4 changed files with 35 additions and 28 deletions
@@ -41,7 +41,7 @@ void dcnV2CudaForward(cublasStatus_t stat, cublasHandle_t handle,
                         const int stride_h, const int stride_w,
                         const int pad_h, const int pad_w,
                         const int dilation_h, const int dilation_w,
-                         const int deformable_group,
+                         const int deformable_group, const int batch_id,
                         const int in_n, const int in_c, const int in_h, const int in_w,
                         const int out_n, const int out_c, const int out_h, const int out_w,
                         const int dst_dim, cudaStream_t stream = cudaStream_t(0));