Skip to content

GitLab

Unverified Commit 561a014b authored by Mikhail Lobanov's avatar Mikhail Lobanov Committed by GitHub
Browse files

Fix Tensor::data<> deprecation. (#2028)

parent c6c3294d
Show whitespace changes
Inline Side-by-side
Showing with 32 additions and 32 deletions
torchvision/csrc/cpu/PSROIAlign_cpu.cpp
View file @ 561a014b
......@@ -340,7 +340,7 @@ std::tuple<at::Tensor, at::Tensor> PSROIAlign_forward_cpu(
input.scalar_type(), "PSROIAlign_forward", [&] {
PSROIAlignForwardCPU<scalar_t>(
output_size,
input.contiguous().data<scalar_t>(),
input.contiguous().data_ptr<scalar_t>(),
spatial_scale,
channels,
height,
......@@ -348,10 +348,10 @@ std::tuple<at::Tensor, at::Tensor> PSROIAlign_forward_cpu(
pooled_height,
pooled_width,
sampling_ratio,
rois.contiguous().data<scalar_t>(),
rois.contiguous().data_ptr<scalar_t>(),
channels_out,
output.data<scalar_t>(),
channel_mapping.data<int>());
output.data_ptr<scalar_t>(),
channel_mapping.data_ptr<int>());
});
return std::make_tuple(output, channel_mapping);
}
......@@ -396,8 +396,8 @@ at::Tensor PSROIAlign_backward_cpu(
grad.scalar_type(), "PSROIAlign_backward", [&] {
PSROIAlignBackwardCPU<scalar_t>(
grad.numel(),
grad.contiguous().data<scalar_t>(),
channel_mapping.data<int>(),
grad.contiguous().data_ptr<scalar_t>(),
channel_mapping.data_ptr<int>(),
num_rois,
spatial_scale,
channels,
......@@ -407,8 +407,8 @@ at::Tensor PSROIAlign_backward_cpu(
pooled_width,
sampling_ratio,
channels_out,
grad_input.data<scalar_t>(),
rois.contiguous().data<scalar_t>());
grad_input.data_ptr<scalar_t>(),
rois.contiguous().data_ptr<scalar_t>());
});
return grad_input;
}
torchvision/csrc/cpu/PSROIPool_cpu.cpp
View file @ 561a014b
......@@ -181,18 +181,18 @@ std::tuple<at::Tensor, at::Tensor> PSROIPool_forward_cpu(
AT_DISPATCH_FLOATING_TYPES_AND_HALF(
input.scalar_type(), "PSROIPool_forward", [&] {
PSROIPoolForward<scalar_t>(
input.contiguous().data<scalar_t>(),
input.contiguous().data_ptr<scalar_t>(),
spatial_scale,
channels,
height,
width,
pooled_height,
pooled_width,
rois.contiguous().data<scalar_t>(),
rois.contiguous().data_ptr<scalar_t>(),
channels_out,
num_rois,
output.data<scalar_t>(),
channel_mapping.data<int>());
output.data_ptr<scalar_t>(),
channel_mapping.data_ptr<int>());
});
return std::make_tuple(output, channel_mapping);
}
......@@ -235,8 +235,8 @@ at::Tensor PSROIPool_backward_cpu(
AT_DISPATCH_FLOATING_TYPES_AND_HALF(
grad.scalar_type(), "PSROIPool_backward", [&] {
PSROIPoolBackward<scalar_t>(
grad.contiguous().data<scalar_t>(),
channel_mapping.data<int>(),
grad.contiguous().data_ptr<scalar_t>(),
channel_mapping.data_ptr<int>(),
num_rois,
spatial_scale,
channels,
......@@ -245,8 +245,8 @@ at::Tensor PSROIPool_backward_cpu(
pooled_height,
pooled_width,
channels_out,
grad_input.data<scalar_t>(),
rois.contiguous().data<scalar_t>());
grad_input.data_ptr<scalar_t>(),
rois.contiguous().data_ptr<scalar_t>());
});
return grad_input;
}
torchvision/csrc/cuda/PSROIAlign_cuda.cu
View file @ 561a014b
......@@ -346,7 +346,7 @@ std::tuple<at::Tensor, at::Tensor> PSROIAlign_forward_cuda(
input.scalar_type(), "PSROIAlign_forward", [&] {
PSROIAlignForwardCUDA<scalar_t><<<grid, block, 0, stream>>>(
output_size,
input.contiguous().data<scalar_t>(),
input.contiguous().data_ptr<scalar_t>(),
spatial_scale,
channels,
height,
......@@ -354,10 +354,10 @@ std::tuple<at::Tensor, at::Tensor> PSROIAlign_forward_cuda(
pooled_height,
pooled_width,
sampling_ratio,
rois.contiguous().data<scalar_t>(),
rois.contiguous().data_ptr<scalar_t>(),
channels_out,
output.data<scalar_t>(),
channel_mapping.data<int>());
output.data_ptr<scalar_t>(),
channel_mapping.data_ptr<int>());
});
AT_CUDA_CHECK(cudaGetLastError());
cudaDeviceSynchronize();
......@@ -416,8 +416,8 @@ at::Tensor PSROIAlign_backward_cuda(
grad.scalar_type(), "PSROIAlign_backward", [&] {
PSROIAlignBackwardCUDA<scalar_t><<<grid, block, 0, stream>>>(
grad.numel(),
grad.contiguous().data<scalar_t>(),
channel_mapping.data<int>(),
grad.contiguous().data_ptr<scalar_t>(),
channel_mapping.data_ptr<int>(),
num_rois,
spatial_scale,
channels,
......@@ -427,8 +427,8 @@ at::Tensor PSROIAlign_backward_cuda(
pooled_width,
sampling_ratio,
channels_out,
grad_input.data<scalar_t>(),
rois.contiguous().data<scalar_t>());
grad_input.data_ptr<scalar_t>(),
rois.contiguous().data_ptr<scalar_t>());
});
AT_CUDA_CHECK(cudaGetLastError());
return grad_input;
......
torchvision/csrc/cuda/PSROIPool_cuda.cu
View file @ 561a014b
......@@ -183,17 +183,17 @@ std::tuple<at::Tensor, at::Tensor> PSROIPool_forward_cuda(
input.scalar_type(), "PSROIPool_forward", [&] {
PSROIPoolForward<scalar_t><<<grid, block, 0, stream>>>(
output_size,
input.contiguous().data<scalar_t>(),
input.contiguous().data_ptr<scalar_t>(),
spatial_scale,
channels,
height,
width,
pooled_height,
pooled_width,
rois.contiguous().data<scalar_t>(),
rois.contiguous().data_ptr<scalar_t>(),
channels_out,
output.data<scalar_t>(),
channel_mapping.data<int>());
output.data_ptr<scalar_t>(),
channel_mapping.data_ptr<int>());
});
AT_CUDA_CHECK(cudaGetLastError());
return std::make_tuple(output, channel_mapping);
......@@ -250,8 +250,8 @@ at::Tensor PSROIPool_backward_cuda(
grad.scalar_type(), "PSROIPool_backward", [&] {
PSROIPoolBackward<scalar_t><<<grid, block, 0, stream>>>(
grad.numel(),
grad.contiguous().data<scalar_t>(),
channel_mapping.data<int>(),
grad.contiguous().data_ptr<scalar_t>(),
channel_mapping.data_ptr<int>(),
num_rois,
spatial_scale,
channels,
......@@ -260,8 +260,8 @@ at::Tensor PSROIPool_backward_cuda(
pooled_height,
pooled_width,
channels_out,
grad_input.data<scalar_t>(),
rois.contiguous().data<scalar_t>());
grad_input.data_ptr<scalar_t>(),
rois.contiguous().data_ptr<scalar_t>());
});
AT_CUDA_CHECK(cudaGetLastError());
return grad_input;
......
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