diff --git a/pytorch3d/csrc/gather_scatter/gather_scatter.h b/pytorch3d/csrc/gather_scatter/gather_scatter.h index 9ab9574f..21d5f7b1 100644 --- a/pytorch3d/csrc/gather_scatter/gather_scatter.h +++ b/pytorch3d/csrc/gather_scatter/gather_scatter.h @@ -53,5 +53,7 @@ at::Tensor GatherScatter( AT_ERROR("Not compiled with GPU support."); #endif } + CHECK_CPU(input); + CHECK_CPU(edges); return GatherScatterCpu(input, edges, directed, backward); } diff --git a/pytorch3d/csrc/interp_face_attrs/interp_face_attrs.h b/pytorch3d/csrc/interp_face_attrs/interp_face_attrs.h index 5ba14462..42f93230 100644 --- a/pytorch3d/csrc/interp_face_attrs/interp_face_attrs.h +++ b/pytorch3d/csrc/interp_face_attrs/interp_face_attrs.h @@ -57,6 +57,8 @@ at::Tensor InterpFaceAttrsForward( AT_ERROR("Not compiled with GPU support."); #endif } + CHECK_CPU(face_attrs); + CHECK_CPU(barycentric_coords); return InterpFaceAttrsForwardCpu(pix_to_face, barycentric_coords, face_attrs); } @@ -106,6 +108,9 @@ std::tuple InterpFaceAttrsBackward( AT_ERROR("Not compiled with GPU support."); #endif } + CHECK_CPU(face_attrs); + CHECK_CPU(barycentric_coords); + CHECK_CPU(grad_pix_attrs); return InterpFaceAttrsBackwardCpu( pix_to_face, barycentric_coords, face_attrs, grad_pix_attrs); } diff --git a/pytorch3d/csrc/iou_box3d/iou_box3d.h b/pytorch3d/csrc/iou_box3d/iou_box3d.h index 84f752b0..d5034b3d 100644 --- a/pytorch3d/csrc/iou_box3d/iou_box3d.h +++ b/pytorch3d/csrc/iou_box3d/iou_box3d.h @@ -44,5 +44,7 @@ inline std::tuple IoUBox3D( AT_ERROR("Not compiled with GPU support."); #endif } + CHECK_CPU(boxes1); + CHECK_CPU(boxes2); return IoUBox3DCpu(boxes1.contiguous(), boxes2.contiguous()); } diff --git a/pytorch3d/csrc/knn/knn.h b/pytorch3d/csrc/knn/knn.h index 7fc8d488..63f204cd 100644 --- a/pytorch3d/csrc/knn/knn.h +++ b/pytorch3d/csrc/knn/knn.h @@ -74,6 +74,8 @@ std::tuple KNearestNeighborIdx( AT_ERROR("Not compiled with GPU support."); #endif } + CHECK_CPU(p1); + CHECK_CPU(p2); return KNearestNeighborIdxCpu(p1, p2, lengths1, lengths2, norm, K); } @@ -140,6 +142,8 @@ std::tuple KNearestNeighborBackward( AT_ERROR("Not compiled with GPU support."); #endif } + CHECK_CPU(p1); + CHECK_CPU(p2); return KNearestNeighborBackwardCpu( p1, p2, lengths1, lengths2, idxs, norm, grad_dists); } diff --git a/pytorch3d/csrc/marching_cubes/marching_cubes.h b/pytorch3d/csrc/marching_cubes/marching_cubes.h index 51c660b1..7a425c07 100644 --- a/pytorch3d/csrc/marching_cubes/marching_cubes.h +++ b/pytorch3d/csrc/marching_cubes/marching_cubes.h @@ -58,5 +58,6 @@ inline std::tuple MarchingCubes( AT_ERROR("Not compiled with GPU support."); #endif } + CHECK_CPU(vol); return MarchingCubesCpu(vol.contiguous(), isolevel); }