Skip to content

feat(kda): enable FlashKDA on SM100 - #124

Merged
icavan merged 4 commits into
mainfrom
codex/flashkda-sm100
Aug 29, 2026
Merged

feat(kda): enable FlashKDA on SM100#124
icavan merged 4 commits into
mainfrom
codex/flashkda-sm100

Conversation

@icavan

@icavan icavan commented Aug 29, 2026

Copy link
Copy Markdown
Collaborator

Summary

  • allow the SM90-derived FlashKDA CuTeDSL path on validated SM100 devices
  • share one architecture guard between backend selection and the direct FlashKDA API
  • keep SM103 disabled until it receives equivalent hardware validation
  • document that this is the compatibility path, not the separate SM100-native modular implementation
  • add architecture unit coverage and an SM100 public-dispatch correctness test

Validation

  • pre-commit run --all-files
  • GB200 SM100 on CUDA device 1: python -m pytest -q tests/test_kda_flashkda_arch.py tests/test_kda_flashkda_sm100.py -s
    • 6 passed
    • the SM100 test calls public cula.kda.kda_prefill, verifies auto intracard CP enters the FlashKDA CP driver, and compares output plus final state against FLA

Scope

This PR only changes architecture admission. The kernel pipeline remains the existing SM90-derived FlashKDA K1/K2 and intracard-CP implementation, which was compiled and executed successfully on SM100. It does not route FlashKDA to the newer SM100-native modular kernels.

Signed-off-by: shuyan.ycf <shuyan.ycf@antgroup.com>
@icavan
icavan requested a review from cherhh August 29, 2026 06:49

@cherhh cherhh left a comment

Copy link
Copy Markdown
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM.

icavan added 3 commits August 29, 2026 15:06
Signed-off-by: shuyan.ycf <shuyan.ycf@antgroup.com>
Signed-off-by: shuyan.ycf <shuyan.ycf@antgroup.com>
Signed-off-by: shuyan.ycf <shuyan.ycf@antgroup.com>
@icavan
icavan merged commit 442f18e into main Aug 29, 2026
2 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants