Skip to content

[JIT-kernel] Add unit test for nsa indexer fused_store_k_cache#19389

Merged
BBuf merged 1 commit intosgl-project:mainfrom
antgroup:ut_nsa_fused_store_k_cache
Mar 2, 2026
Merged

[JIT-kernel] Add unit test for nsa indexer fused_store_k_cache#19389
BBuf merged 1 commit intosgl-project:mainfrom
antgroup:ut_nsa_fused_store_k_cache

Conversation

@yuan-luo
Copy link
Copy Markdown
Collaborator

@yuan-luo yuan-luo commented Feb 26, 2026 •
edited
Loading

Motivation

Follow up #19148 review comments, introduce a unit test for this jit kernel.

➜  sglang_dev3 git:(support_nsa_fuse_store_k_cache) ✗ python -m pytest -q -rs python/sglang/jit_kernel/tests/test_fused_store_index_cache.py
..............                                                                                                                                                                                                                              [100%]
================================================================================================================ warnings summary =================================================================================================================
<frozen importlib._bootstrap>:488
  <frozen importlib._bootstrap>:488: DeprecationWarning: builtin type SwigPyPacked has no __module__ attribute

<frozen importlib._bootstrap>:488
  <frozen importlib._bootstrap>:488: DeprecationWarning: builtin type SwigPyObject has no __module__ attribute

-- Docs: https://docs.pytest.org/en/stable/how-to/capture-warnings.html
14 passed, 2 warnings in 34.76s

Modifications

Accuracy Tests

Benchmarking and Profiling

Checklist

Review Process

  1. Ping Merge Oncalls to start the PR flow. See the PR Merge Process.
  2. Get approvals from CODEOWNERS and other reviewers.
  3. Trigger CI tests with comments or contact authorized users to do so.
    • /tag-run-ci-label, /rerun-failed-ci, /tag-and-rerun-ci
  4. After green CI and required approvals, ask Merge Oncalls to merge.

@gemini-code-assist
Copy link
Copy Markdown
Contributor

gemini-code-assist Bot commented Feb 26, 2026

Warning

You have reached your daily quota limit. Please wait up to 24 hours and I will start processing your requests again!

@yuan-luo yuan-luo requested a review from Fridge003 February 26, 2026 03:32
@yuan-luo yuan-luo mentioned this pull request Feb 26, 2026
Merged
5 tasks
@yuan-luo
Copy link
Copy Markdown
Collaborator Author

yuan-luo commented Feb 26, 2026

/rerun-failed-ci

@yuan-luo
Copy link
Copy Markdown
Collaborator Author

yuan-luo commented Feb 28, 2026

/tag-and-rerun-ci

@yuan-luo
Copy link
Copy Markdown
Collaborator Author

yuan-luo commented Feb 28, 2026

/rerun-failed-ci

@yuan-luo
Copy link
Copy Markdown
Collaborator Author

yuan-luo commented Mar 2, 2026

/rerun-failed-ci

@BBuf BBuf merged commit f6ee6dc into sgl-project:main Mar 2, 2026
360 of 412 checks passed
@yuan-luo yuan-luo deleted the ut_nsa_fused_store_k_cache branch March 2, 2026 04:40
Kangyan-Zhou pushed a commit to Kangyan-Zhou/sglang that referenced this pull request Mar 4, 2026
@yuan-luo @Kangyan-Zhou
[JIT-kernel] Add unit test for nsa indexer fused_store_k_cache (sgl-p…
77c56a7 
…roject#19389)

Co-authored-by: luoyuan.luo <luoyuan.luo@antgroup.com>
magicYang1573 pushed a commit to magicYang1573/sglang that referenced this pull request Mar 9, 2026
@yuan-luo
[JIT-kernel] Add unit test for nsa indexer fused_store_k_cache (sgl-p…
8d0e8d7 
…roject#19389)

Co-authored-by: luoyuan.luo <luoyuan.luo@antgroup.com>
Wangzheee pushed a commit to Wangzheee/sglang that referenced this pull request Mar 21, 2026
@yuan-luo @Wangzheee
[JIT-kernel] Add unit test for nsa indexer fused_store_k_cache (sgl-p…
e58bba2 
…roject#19389)

Co-authored-by: luoyuan.luo <luoyuan.luo@antgroup.com>
JustinTong0323 pushed a commit to JustinTong0323/sglang that referenced this pull request Apr 7, 2026
@yuan-luo
[JIT-kernel] Add unit test for nsa indexer fused_store_k_cache (sgl-p…
cfdd9e9 
…roject#19389)

Co-authored-by: luoyuan.luo <luoyuan.luo@antgroup.com>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Reviewers

@BBuf BBuf BBuf approved these changes

@DarkSharpness DarkSharpness Awaiting requested review from DarkSharpness DarkSharpness is a code owner

@celve celve Awaiting requested review from celve celve is a code owner

@HydraQYH HydraQYH Awaiting requested review from HydraQYH HydraQYH is a code owner

@Fridge003 Fridge003 Awaiting requested review from Fridge003

Assignees

No one assigned

Labels

run-ci

Projects

None yet

Milestone

No milestone

Development

Successfully merging this pull request may close these issues.

2 participants

@yuan-luo @BBuf