colossalai/legacy/utils/activation_checkpoint.py · 3acbf6d4968e0559629f0d6d317e5bac41ad5df0 · OpenDAS / ColossalAI · GitLab

Find file Blame History Permalink

[npu] add npu support for hybrid plugin and llama (#5090) · 3acbf6d4
Xuanlei Zhao authored Nov 22, 2023
```
* llama 3d

* update

* fix autocast
```
3acbf6d4

activation_checkpoint.py 9.58 KB