diff --git a/pytorch3d/csrc/sample_pdf/sample_pdf_cpu.cpp b/pytorch3d/csrc/sample_pdf/sample_pdf_cpu.cpp index c3300520..db19f386 100644 --- a/pytorch3d/csrc/sample_pdf/sample_pdf_cpu.cpp +++ b/pytorch3d/csrc/sample_pdf/sample_pdf_cpu.cpp @@ -32,8 +32,7 @@ void SamplePdfCpu_worker( auto bins_a = bins.accessor(); auto weights_a = weights.accessor(); - float* __restrict__ output_p = - outputs.data_ptr() + start_batch * n_samples; + float* output_p = outputs.data_ptr() + start_batch * n_samples; #ifdef USE_BINARY_SEARCH std::vector partial_sums(n_bins);