github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-05-04 01:48:43 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
2899cfdabfdf205121c5ebc8a47494db3728c7af
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