ColossalAI/tests/test_tensor/_utils
Jiarui Fang 060b917daf
[refactor] remove gpc dependency in colotensor's _ops (#1189)
2022-07-04 18:54:37 +08:00
..
__init__.py [tensor]add assert for colo_tensor 1Drow (#846) 2022-04-24 14:12:45 +08:00
_util.py [refactor] remove gpc dependency in colotensor's _ops (#1189) 2022-07-04 18:54:37 +08:00