Skip to content

Organize FE OSS tests by feature#372

Merged
hwanseoc merged 1 commit into
NVIDIA:developfrom
hwanseoc:hwanseoc/organize-fe-oss-tests
Jul 10, 2026
Merged

Organize FE OSS tests by feature#372
hwanseoc merged 1 commit into
NVIDIA:developfrom
hwanseoc:hwanseoc/organize-fe-oss-tests

Conversation

@hwanseoc

@hwanseoc hwanseoc commented Jul 9, 2026

Copy link
Copy Markdown
Member

Summary

  • group FE OSS tests into flat, top-level feature folders
  • rename block_sparse_attention/ to bsa/ and add sdpa/, gemm/, grouped_gemm/, and norm/
  • keep existing dsa/ and nsa/ folders unchanged
  • update imports, documentation paths, and integration guidance

Final layout

  • bsa/
  • dsa/
  • nsa/
  • sdpa/
  • gemm/
  • grouped_gemm/
  • norm/

Moved files

From To
test/python/fe_api/block_sparse_attention/__init__.py test/python/fe_api/bsa/__init__.py
test/python/fe_api/block_sparse_attention/bsa_reference.py test/python/fe_api/bsa/bsa_reference.py
test/python/fe_api/block_sparse_attention/bsa_utils.py test/python/fe_api/bsa/bsa_utils.py
test/python/fe_api/block_sparse_attention/test_BSA_attention_backward.py test/python/fe_api/bsa/test_BSA_attention_backward.py
test/python/fe_api/block_sparse_attention/test_BSA_attention_forward.py test/python/fe_api/bsa/test_BSA_attention_forward.py
test/python/fe_api/test_gemm_amax.py test/python/fe_api/gemm/test_gemm_amax.py
test/python/fe_api/test_gemm_amax_utils.py test/python/fe_api/gemm/test_gemm_amax_utils.py
test/python/fe_api/test_gemm_dsrelu.py test/python/fe_api/gemm/test_gemm_dsrelu.py
test/python/fe_api/test_gemm_dsrelu_utils.py test/python/fe_api/gemm/test_gemm_dsrelu_utils.py
test/python/fe_api/test_gemm_srelu.py test/python/fe_api/gemm/test_gemm_srelu.py
test/python/fe_api/test_gemm_srelu_utils.py test/python/fe_api/gemm/test_gemm_srelu_utils.py
test/python/fe_api/test_gemm_swiglu.py test/python/fe_api/gemm/test_gemm_swiglu.py
test/python/fe_api/test_gemm_swiglu_utils.py test/python/fe_api/gemm/test_gemm_swiglu_utils.py
test/python/fe_api/test_discrete_grouped_gemm_dswiglu.py test/python/fe_api/grouped_gemm/test_discrete_grouped_gemm_dswiglu.py
test/python/fe_api/test_discrete_grouped_gemm_dswiglu_utils.py test/python/fe_api/grouped_gemm/test_discrete_grouped_gemm_dswiglu_utils.py
test/python/fe_api/test_discrete_grouped_gemm_swiglu.py test/python/fe_api/grouped_gemm/test_discrete_grouped_gemm_swiglu.py
test/python/fe_api/test_discrete_grouped_gemm_swiglu_utils.py test/python/fe_api/grouped_gemm/test_discrete_grouped_gemm_swiglu_utils.py
test/python/fe_api/test_grouped_gemm_dglu.py test/python/fe_api/grouped_gemm/test_grouped_gemm_dglu.py
test/python/fe_api/test_grouped_gemm_dsrelu.py test/python/fe_api/grouped_gemm/test_grouped_gemm_dsrelu.py
test/python/fe_api/test_grouped_gemm_dsrelu_utils.py test/python/fe_api/grouped_gemm/test_grouped_gemm_dsrelu_utils.py
test/python/fe_api/test_grouped_gemm_dswiglu.py test/python/fe_api/grouped_gemm/test_grouped_gemm_dswiglu.py
test/python/fe_api/test_grouped_gemm_dswiglu_utils.py test/python/fe_api/grouped_gemm/test_grouped_gemm_dswiglu_utils.py
test/python/fe_api/test_grouped_gemm_glu.py test/python/fe_api/grouped_gemm/test_grouped_gemm_glu.py
test/python/fe_api/test_grouped_gemm_glu_hadamard.py test/python/fe_api/grouped_gemm/test_grouped_gemm_glu_hadamard.py
test/python/fe_api/test_grouped_gemm_quant.py test/python/fe_api/grouped_gemm/test_grouped_gemm_quant.py
test/python/fe_api/test_grouped_gemm_quant_utils.py test/python/fe_api/grouped_gemm/test_grouped_gemm_quant_utils.py
test/python/fe_api/test_grouped_gemm_srelu.py test/python/fe_api/grouped_gemm/test_grouped_gemm_srelu.py
test/python/fe_api/test_grouped_gemm_srelu_utils.py test/python/fe_api/grouped_gemm/test_grouped_gemm_srelu_utils.py
test/python/fe_api/test_grouped_gemm_swiglu.py test/python/fe_api/grouped_gemm/test_grouped_gemm_swiglu.py
test/python/fe_api/test_grouped_gemm_swiglu_utils.py test/python/fe_api/grouped_gemm/test_grouped_gemm_swiglu_utils.py
test/python/fe_api/test_grouped_gemm_wgrad.py test/python/fe_api/grouped_gemm/test_grouped_gemm_wgrad.py
test/python/fe_api/test_grouped_gemm_wgrad_utils.py test/python/fe_api/grouped_gemm/test_grouped_gemm_wgrad_utils.py
test/python/fe_api/test_rmsnorm_rht_amax.py test/python/fe_api/norm/test_rmsnorm_rht_amax.py
test/python/fe_api/test_sdpa_bwd.py test/python/fe_api/sdpa/test_sdpa_bwd.py
test/python/fe_api/test_sdpa_bwd_utils.py test/python/fe_api/sdpa/test_sdpa_bwd_utils.py
test/python/fe_api/test_sdpa_fwd.py test/python/fe_api/sdpa/test_sdpa_fwd.py
test/python/fe_api/test_sdpa_fwd_utils.py test/python/fe_api/sdpa/test_sdpa_fwd_utils.py

Additional changes

  • Add test/python/fe_api/README.md documenting the feature layout.
  • Update imports and documentation references to the new paths.

Testing

  • pytest --collect-only -q test/python/fe_api
  • 3900 collected, 2 deselected
  • git diff --check origin/develop...HEAD

@coderabbitai ignore

@hwanseoc
hwanseoc force-pushed the hwanseoc/organize-fe-oss-tests branch from dffbcfe to 401273a Compare July 9, 2026 21:17
@hwanseoc

hwanseoc commented Jul 9, 2026

Copy link
Copy Markdown
Member Author

@cudnn-ci-bot run

@cudnn-ci-bot

Copy link
Copy Markdown

🚀 Running mirror pipeline

Branch: cudnn-gh/pr-372-401273a
Pipeline: 57452060

@hwanseoc
hwanseoc requested a review from Anerudhan July 9, 2026 23:07
@Anerudhan Anerudhan added mod-frontend cuDNN frontend APIs, operation graph construction, plans, and user-facing wrappers. orig-nv-eng Reported or requested by NVIDIA engineering. cat-enhancements labels Jul 10, 2026
@Anerudhan Anerudhan added this to the Frontend 1.27.0 milestone Jul 10, 2026
@hwanseoc
hwanseoc merged commit 1ffbf98 into NVIDIA:develop Jul 10, 2026
1 check passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

cat-enhancements mod-frontend cuDNN frontend APIs, operation graph construction, plans, and user-facing wrappers. orig-nv-eng Reported or requested by NVIDIA engineering.

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants