Skip to content

JIT LTO Cagra Search#1807

Open
divyegala wants to merge 223 commits intorapidsai:mainfrom
divyegala:cagra-search-jit-lto
Open

JIT LTO Cagra Search#1807
divyegala wants to merge 223 commits intorapidsai:mainfrom
divyegala:cagra-search-jit-lto

Conversation

@divyegala
Copy link
Copy Markdown
Member

@divyegala divyegala commented Feb 15, 2026 •
edited
Loading

divyegala added 30 commits October 2, 2025 18:33
@divyegala
jit lto interleaved scan
a024f61
@divyegala
fix dependencies.yaml
45da4aa
@divyegala
generate files at build time, use tags to avoid compilation of types
a7c8621
@divyegala
passing tests
eb2d74b
@divyegala
update gitignore
d2318e8
@divyegala
separate out distance function from main kernel
5e6afcd
@divyegala
fix deps
6eee4da
@divyegala
add filters as jit device functions, rework caching logic
1de8f28
@divyegala
lto post lambda, cleanup files, generate cmake in build dir
84c6020
@divyegala
don't read hardcoded kernels, use generator properly
22680c8
@divyegala
random cmake changes carried over from 25.10
37f1163
@divyegala
cmake format
0ae5383
@divyegala
remove dep on kernel list
fe56aec
@divyegala
attempt to solve overlinking problem
40c8fd6
@divyegala
reorder if-else in compiler check
e87a8c7
@divyegala
Merge branch 'branch-25.12' into jit-lto-ivf-flat-interleaved
179d733
@divyegala
use cudart apis
32a67bd
@divyegala
merge
c27612e
@divyegala
attempt to link cudart
a4b48b1
@divyegala
revert cudart link, try all arch build of jit lto fatbin sources
d5d692e
@divyegala
cmake format
1c6dd94
@divyegala
missing shared mem setting
30f5ab6
@divyegala
separate cuda 12 and 13 compilation
9674969
@divyegala
merge upstream
24fc47d
@divyegala
remove bench
db9a487
@divyegala
c include directory
aa9294f
@divyegala
style check
2eb77fe
@divyegala
merge upstream
6c685fa
@divyegala
guard cuda calls and use shared_ptr
3e35b99
@divyegala
add AlgorithmPlanner to main target
d0ff62c
@achirkin achirkin mentioned this pull request Mar 9, 2026
Merged
@KyleFromNVIDIA KyleFromNVIDIA mentioned this pull request Mar 9, 2026
Merged
KyleFromNVIDIA
KyleFromNVIDIA reviewed Mar 11, 2026
View reviewed changes
Comment thread dependencies.yaml Outdated
KyleFromNVIDIA
KyleFromNVIDIA requested changes Mar 12, 2026
View reviewed changes
Comment thread cpp/src/neighbors/detail/cagra/jit_lto_kernels/compute_distance_kernel.cu.in Outdated
Comment thread cpp/src/neighbors/detail/cagra/jit_lto_kernels/setup_workspace_kernel.cu.in Outdated
KyleFromNVIDIA
KyleFromNVIDIA requested changes Mar 13, 2026
View reviewed changes
Comment thread cpp/src/neighbors/detail/cagra/jit_lto_kernels/setup_workspace_kernel.cu.in Outdated
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Reviewers

@KyleFromNVIDIA KyleFromNVIDIA KyleFromNVIDIA approved these changes

At least 1 approving review is required to merge this pull request.

Assignees

@divyegala divyegala

Labels

DO NOT MERGE feature request New feature or request non-breaking Introduces a non-breaking change

Projects

Unstructured Data Processing
Status: No status

Milestone

No milestone

Development

Successfully merging this pull request may close these issues.

Apply JIT-LTO to CAGRA search kernels

4 participants

@divyegala @KyleFromNVIDIA @dantegd @bdice