ColossalAI/colossalai/utils/tensor_detector/__init__.py at 1bd8a72fc97ca3a5b001d340e65891973b17eccc - ColossalAI - Gitea: Git with a cup of tea at home

github/ColossalAI

mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-07-11 00:56:38 +00:00

Files

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