colossalai/booster/mixed_precision/fp16_torch.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

fp16_torch.py 4.69 KB