Skip to content

Add grid_cluster CUDA kernel#586

Merged
akihironitta merged 5 commits intomasterfrom
port-cluster-02-grid-cuda
Mar 23, 2026
Merged

Add grid_cluster CUDA kernel#586
akihironitta merged 5 commits intomasterfrom
port-cluster-02-grid-cuda

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.
@akihironitta akihironitta force-pushed the port-cluster-02-grid-cuda branch from ce6534c to 31f480b Compare March 16, 2026 07:37
@akihironitta akihironitta force-pushed the port-cluster-01-grid-dispatch branch from 175eca7 to 2c53091 Compare March 16, 2026 07:37
@github-actions github-actions bot removed the ci label Mar 16, 2026
@akihironitta akihironitta force-pushed the port-cluster-01-grid-dispatch branch from 2c53091 to 4949ae5 Compare March 22, 2026 19:36
Base automatically changed from port-cluster-01-grid-dispatch to master March 22, 2026 22:15
@akihironitta akihironitta merged commit e812618 into master Mar 23, 2026
26 checks passed
@akihironitta akihironitta deleted the port-cluster-02-grid-cuda branch March 23, 2026 00:35
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