ColossalAI/tests/test_tensor
HELSON bf5066fba7
[refactor] refactor ColoTensor's unit tests (#1340)
2022-07-19 15:46:24 +08:00
..
common_utils [refactor] refactor ColoTensor's unit tests (#1340) 2022-07-19 15:46:24 +08:00
core [refactor] refactor ColoTensor's unit tests (#1340) 2022-07-19 15:46:24 +08:00
model [refactor] refactor ColoTensor's unit tests (#1340) 2022-07-19 15:46:24 +08:00
ops [refactor] refactor ColoTensor's unit tests (#1340) 2022-07-19 15:46:24 +08:00
test_chunk.py [hotfix] ZeroDDP use new process group (#1333) 2022-07-18 14:14:52 +08:00
test_context.py [refactory] add nn.parallel module (#1068) 2022-06-06 15:34:41 +08:00
test_parameter.py [refactor] refactor ColoTensor's unit tests (#1340) 2022-07-19 15:46:24 +08:00
test_zero_optim.py [refactor] refactor ColoTensor's unit tests (#1340) 2022-07-19 15:46:24 +08:00