Skip to content

Comments

Align softmax_online scaffold with issue #38#42

Merged
debashishc merged 1 commit intomainfrom
tmp/main-online-softmax
Feb 8, 2026
Merged

Align softmax_online scaffold with issue #38#42
debashishc merged 1 commit intomainfrom
tmp/main-online-softmax

Conversation

@debashishc
Copy link
Contributor

@debashishc debashishc commented Feb 8, 2026

Summary

  • switch softmax_online to backend registry flow (ref + kernel) with simpler dispatch
  • add kernel module from PR26 as explicit non-production reference (with credit header)
  • enforce strict dim=-1 for softmax_online op/ref paths
  • exercise kernel backend in tests (forward + backward coverage)
  • align softmax benchmarks with backend selection APIs instead of env mode

Validation

  • uv run ruff check forge_cute_py/ops/softmax_online.py forge_cute_py/ref/softmax_online.py tests/test_softmax_online.py
  • uv run pytest tests/test_softmax_online.py -q
  • uv run pytest -q

@debashishc debashishc merged commit 2fe8f4f into main Feb 8, 2026
2 checks passed
@debashishc debashishc deleted the tmp/main-online-softmax branch February 8, 2026 14:26
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.

1 participant