Skip to main content

Module ptx_kernels

Module ptx_kernels 

Source
Expand description

GPU PTX kernels for graph algorithm operations.

Each kernel is emitted as a self-contained PTX module string, parameterised on SM version. PTX ISA is selected by SM: SM>=100 -> 8.7 (Blackwell), SM>=90 -> 8.4 (Hopper), SM>=80 -> 8.0 (Ampere), else -> 7.5 (Turing).

IMPORTANT: PTX kernel bodies use string concatenation (NOT format!()) for sections containing %rd, %r, %f, %fd register names, which Rust’s format macro would misinterpret as unused format arguments in edition 2024.

Functions§

bfs_level_ptx
Frontier-based BFS single-step level update.
community_label_ptx
Label propagation single step.
csr_spmv_bool_ptx
Boolean CSR sparse mat-vec (for matrix-form BFS).
dijkstra_relax_ptx
Dijkstra single edge relaxation kernel.
fw_inner_ptx
Floyd-Warshall inner DP update.
pagerank_step_ptx
PageRank single power-iteration step.
triangle_count_ptx
Triangle counting per row.