github
/
ColossalAI
mirror of https://github.com/hpcaitech/ColossalAI
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
ColossalAI/colossalai/utils/tensor_detector/__init__.py

1 line
43 B
Python
Raw Normal View History

Added tensor detector (#393) * Added tensor detector * Added the - states * Allowed change include_cpu when detect()
2022-03-14 10:01:46 +00:00
from .tensor_detector import TensorDetector