diff --git a/torchvision/csrc/ops/quantized/cpu/qroi_align_kernel.cpp b/torchvision/csrc/ops/quantized/cpu/qroi_align_kernel.cpp index e34b277747e..cfd5ec4ee97 100644 --- a/torchvision/csrc/ops/quantized/cpu/qroi_align_kernel.cpp +++ b/torchvision/csrc/ops/quantized/cpu/qroi_align_kernel.cpp @@ -23,11 +23,15 @@ void qroi_align_forward_kernel_impl( bool aligned, const at::Tensor& t_rois, T* output) { - const T* input = t_input.contiguous().data_ptr(); + // Don't delete these otherwise the .data_ptr() data might be undefined + auto t_input_cont = t_input.contiguous(); + auto t_rois_cont = t_rois.contiguous(); + + const T* input = t_input_cont.data_ptr(); int64_t input_zp = t_input.q_zero_point(); float input_scale = t_input.q_scale(); - const T* rois = t_rois.contiguous().data_ptr(); + const T* rois = t_rois_cont.data_ptr(); int64_t rois_zp = t_rois.q_zero_point(); float rois_scale = t_rois.q_scale();