diff options
| author | Jerry Zhang <jerryzh@fb.com> | 2019-01-15 18:39:28 -0800 |
|---|---|---|
| committer | Facebook Github Bot <facebook-github-bot@users.noreply.github.com> | 2019-01-15 18:42:02 -0800 |
| commit | 5e72e99c86b22421946ee6cc9b1bef72d9ec6de9 (patch) | |
| tree | 5e7905aa84eee6ce015dabf0d063ded2d26b3b88 /modules | |
| parent | 8b5894491c0064d24268a584246ef0496bd5100b (diff) | |
| download | pytorch-5e72e99c86b22421946ee6cc9b1bef72d9ec6de9.tar.gz pytorch-5e72e99c86b22421946ee6cc9b1bef72d9ec6de9.tar.bz2 pytorch-5e72e99c86b22421946ee6cc9b1bef72d9ec6de9.zip | |
Remaining Tensor API fixes - dims() -> sizes() (#15743)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/15743
Remaining fixes so that D12812029 will compile
Reviewed By: dzhulgakov
Differential Revision: D13535559
fbshipit-source-id: 2c8b3403570c8c35ac8efe2d827233abc0e6e0d1
Diffstat (limited to 'modules')
| -rw-r--r-- | modules/detectron/ps_roi_pool_op.cu | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/modules/detectron/ps_roi_pool_op.cu b/modules/detectron/ps_roi_pool_op.cu index c99c3e370b..52e0a3129c 100644 --- a/modules/detectron/ps_roi_pool_op.cu +++ b/modules/detectron/ps_roi_pool_op.cu @@ -246,7 +246,7 @@ bool PSRoIPoolOp<float, CUDAContext>::RunOnDevice() { auto* Y = Output(0, {R.dim32(0), output_dim_, pooled_height_, pooled_width_}, at::dtype<float>()); // PSRoI pooled data auto* A = Output(1, Y->sizes(), at::dtype<int>()); // mapping_channel - int output_size = Y->size(); + int output_size = Y->numel(); PSRoIPoolForward<float><<<CAFFE_GET_BLOCKS(output_size), CAFFE_CUDA_NUM_THREADS, 0, context_.cuda_stream()>>>( |