feat: LoRA kernel support for bias, dropout, dora, embeddings (#3528) [skip ci]

Browse Source

* feat: LoRA kernel support for bias, dropout, dora, embeddings

* chore: lint

* chore: lint

* address PR feedback, add regression tests, add fsdp2 tests for lora kernels

* update tests for new sigs

* update tests now that bias and dropout are supported

...

This commit is contained in:

Wing Lian

2026-03-22 13:53:19 -04:00

committed by GitHub

parent a67392c427

commit b3289fd190

13 changed files with 2847 additions and 448 deletions

Show all changes Ignore whitespace when comparing lines Ignore changes in amount of whitespace Ignore changes in whitespace at EOL

Download Patch File Download Diff File Expand all files Collapse all files

1245

tests/e2e/kernels/test_lora_features.py Normal file

File diff suppressed because it is too large Load Diff