pytorch3d
pytorch3d copied to clipboard
Add fp16 support for sample_farthest_points, ball_query and knn
add fp16 CUDA supports for sample_farthest_points, ball_query functions via AT_DISPATCH_FLOATING_TYPES_AND_HALF and knn