ColossalAI/tests/test_tensor/common_utils
HELSON 7a8702c06d
[colotensor] add Tensor.view op and its unit test (#1343)
[colotensor] add megatron initialization for gpt2
2022-07-21 10:53:15 +08:00
..
__init__.py
_utils.py [colotensor] add Tensor.view op and its unit test (#1343) 2022-07-21 10:53:15 +08:00