github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-07-12 19:06:17 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
0e616462a7f9e8faaa33d1700a2020ceb03ccd34
ColossalAI/colossalai/utils/device.py
Xuanlei Zhao 3acbf6d496 [npu] add npu support for hybrid plugin and llama (#5090) 
* llama 3d

* update

* fix autocast
2023-11-22 19:23:21 +08:00

5.5 KiB
Raw Blame History

View Raw
Reference in New Issue View Git Blame Copy Permalink