Files
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

from .tensor_detector import TensorDetector