colossalai/kernel/cuda_native/scaled_softmax.py · 7bc5a8e338a1240070ca82f96721c451060f1a70 · OpenDAS / ColossalAI · GitLab

"examples/language/llama2/test_ci.sh" did not exist on "b0ce5a10326912961f0bc07cbbd250bab7b9c399"

Find file Blame History Permalink

[kernel] added kernel loader to softmax autograd function (#3093) · 95a36eae
Frank Lee authored Mar 10, 2023
```
* [kernel] added kernel loader to softmax autograd function

* [release] v0.2.6
```
95a36eae

scaled_softmax.py 6.56 KB