diff --git a/pytorch3d/csrc/pulsar/include/renderer.forward.device.h b/pytorch3d/csrc/pulsar/include/renderer.forward.device.h index 7e2a11f3..3f0412f5 100644 --- a/pytorch3d/csrc/pulsar/include/renderer.forward.device.h +++ b/pytorch3d/csrc/pulsar/include/renderer.forward.device.h @@ -102,6 +102,7 @@ void forward( self->workspace_d, self->workspace_size, stream); + CHECKLAUNCH(); SORT_ASCENDING_WS( self->min_depth_d, self->min_depth_sorted_d, @@ -111,6 +112,7 @@ void forward( self->workspace_d, self->workspace_size, stream); + CHECKLAUNCH(); SORT_ASCENDING_WS( self->min_depth_d, self->min_depth_sorted_d,