Commit Graph

6 Commits (0d212183c47cbc09f9d7cb32006057bc3694e370)

Author SHA1 Message Date
Jiarui Fang 4a76084dc9
[tensor] add zero_like colo op, important for Optimizer (#1236)
2 years ago
Jiarui Fang a98319f023
[tensor] torch function return colotensor (#1229)
2 years ago
Jiarui Fang ae7d3f4927
[refactor] move process group from _DistSpec to ColoTensor. (#1203)
2 years ago
Jiarui Fang 4b9bba8116
[ColoTensor] rename APIs and add output_replicate to ComputeSpec (#1168)
2 years ago
ver217 ae86151968
[tensor] add more element-wise ops (#1155)
2 years ago
Jiarui Fang a00644079e
reorgnize colotensor directory (#1062)
3 years ago