From 3b0fbb2dbd4abc5dcbd903647df3b3d1be85777f Mon Sep 17 00:00:00 2001 From: Tim Date: Thu, 11 Apr 2019 17:09:43 -0400 Subject: [PATCH] add type-cast to fix building errors on windows --- lib/model/csrc/cuda/ROIAlign_cuda.cu | 4 ++-- lib/model/csrc/cuda/ROIPool_cuda.cu | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/lib/model/csrc/cuda/ROIAlign_cuda.cu b/lib/model/csrc/cuda/ROIAlign_cuda.cu index 5fe97ca90..1142fb375 100644 --- a/lib/model/csrc/cuda/ROIAlign_cuda.cu +++ b/lib/model/csrc/cuda/ROIAlign_cuda.cu @@ -272,7 +272,7 @@ at::Tensor ROIAlign_forward_cuda(const at::Tensor& input, auto output_size = num_rois * pooled_height * pooled_width * channels; cudaStream_t stream = at::cuda::getCurrentCUDAStream(); - dim3 grid(std::min(THCCeilDiv(output_size, 512L), 4096L)); + dim3 grid(std::min(THCCeilDiv((long)output_size, 512L), 4096L)); dim3 block(512); if (output.numel() == 0) { @@ -317,7 +317,7 @@ at::Tensor ROIAlign_backward_cuda(const at::Tensor& grad, cudaStream_t stream = at::cuda::getCurrentCUDAStream(); - dim3 grid(std::min(THCCeilDiv(grad.numel(), 512L), 4096L)); + dim3 grid(std::min(THCCeilDiv((long)grad.numel(), 512L), 4096L)); dim3 block(512); // handle possibly empty gradients diff --git a/lib/model/csrc/cuda/ROIPool_cuda.cu b/lib/model/csrc/cuda/ROIPool_cuda.cu index b826dd9bc..8f072ffc2 100644 --- a/lib/model/csrc/cuda/ROIPool_cuda.cu +++ b/lib/model/csrc/cuda/ROIPool_cuda.cu @@ -126,7 +126,7 @@ std::tuple ROIPool_forward_cuda(const at::Tensor& input, cudaStream_t stream = at::cuda::getCurrentCUDAStream(); - dim3 grid(std::min(THCCeilDiv(output_size, 512L), 4096L)); + dim3 grid(std::min(THCCeilDiv((long)output_size, 512L), 4096L)); dim3 block(512); if (output.numel() == 0) { @@ -173,7 +173,7 @@ at::Tensor ROIPool_backward_cuda(const at::Tensor& grad, cudaStream_t stream = at::cuda::getCurrentCUDAStream(); - dim3 grid(std::min(THCCeilDiv(grad.numel(), 512L), 4096L)); + dim3 grid(std::min(THCCeilDiv((long)grad.numel(), 512L), 4096L)); dim3 block(512); // handle possibly empty gradients