github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-01-20 17:44:17 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
46c6cc79a90bf578cb92d8d2ab5b9e2ff0e818fd
ColossalAI/colossalai/utils/tensor_detector/__init__.py
LuGY a9c27be42e Added tensor detector (#393) 
* Added tensor detector

* Added the - states

* Allowed change include_cpu when detect()
2022-03-14 18:01:46 +08:00

1 line
43 B
Python
Raw Blame History

from .tensor_detector import TensorDetector
Reference in New Issue View Git Blame Copy Permalink