You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
ColossalAI/colossalai/tensor
ver217 7faef93326
fix dist spec mgr (#1045)
3 years ago
..
_ops [tensor] refactor parallel action (#1007) 3 years ago
graph
modules [Tensor] add module check and bert test (#1031) 3 years ago
optim
__init__.py [tensor] ColoTensor supports ZeRo (#1015) 3 years ago
chunk.py [tensor] ColoTensor supports ZeRo (#1015) 3 years ago
colo_parameter.py [tensor] ColoTensor supports ZeRo (#1015) 3 years ago
colo_tensor.py [tensor] refactor parallel action (#1007) 3 years ago
const.py
dist_spec_mgr.py fix dist spec mgr (#1045) 3 years ago
distspec.py [Tensor] add module check and bert test (#1031) 3 years ago
module_utils.py [Tensor] add module check and bert test (#1031) 3 years ago
op_wrapper.py
param_op_hook.py [tensor] ColoTensor supports ZeRo (#1015) 3 years ago
spec.py [tensor] refactor parallel action (#1007) 3 years ago
utils.py