Commit Graph

5 Commits (3c3714fc2a0ea9b68c57ab96d871cbb948fb345c)

Author SHA1 Message Date
Jiarui Fang 3ce4463fe6
[utils] remove lazy_memory_allocate from ColoInitContext (#1844)
2 years ago
Jiarui Fang 49832b2344
[refactory] add nn.parallel module (#1068)
3 years ago
ver217 ad536e308e
[tensor] refactor colo-tensor (#992)
3 years ago
Jiarui Fang d01d3b8cb0
colo init context add device attr. (#866)
3 years ago
Jiarui Fang 8789850eea
Init Conext supports lazy allocate model memory (#842)
3 years ago