Skip to content

Add fps dispatch + CPU kernel#587

Merged
akihironitta merged 10 commits intomasterfrom
port-cluster-03-fps-dispatch
Mar 23, 2026
Merged

Add fps dispatch + CPU kernel#587
akihironitta merged 10 commits intomasterfrom
port-cluster-03-fps-dispatch

Conversation

@akihironitta
Copy link
Member

No description provided.

Port grid voxelization from pytorch_cluster into pyg-lib.
Each point is assigned a 1D cluster index based on its quantized
voxel position: floor((pos - start) / size), then flattened
with cumulative voxel counts.
Port CUDA grid voxelization kernel. One thread per point computes
the flattened voxel index. Tests verify CPU/CUDA parity.
Port greedy farthest point sampling from pytorch_cluster.
Uses at::parallel_for over batch dimension. Iteratively selects
the point farthest from the already-selected set.
@akihironitta akihironitta force-pushed the port-cluster-03-fps-dispatch branch from b5dd5d5 to a9f95f8 Compare March 16, 2026 07:37
@akihironitta akihironitta force-pushed the port-cluster-02-grid-cuda branch from ce6534c to 31f480b Compare March 16, 2026 07:37
@github-actions github-actions bot removed the ci label Mar 16, 2026
Base automatically changed from port-cluster-02-grid-cuda to master March 23, 2026 00:35
@akihironitta akihironitta merged commit b99c438 into master Mar 23, 2026
29 checks passed
@akihironitta akihironitta deleted the port-cluster-03-fps-dispatch branch March 23, 2026 01:01
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant