ColossalAI/colossalai/tensor/_ops
ver217 c2fdc6a011
[tensor] derive compute pattern from dist spec (#971)
* derive compute pattern from dist spec

* polish code
2022-05-16 14:58:08 +08:00
..
__init__.py add DistSpec for loss and test_model (#947) 2022-05-13 20:29:50 +08:00
addmm.py [tensor] derive compute pattern from dist spec (#971) 2022-05-16 14:58:08 +08:00
element_wise.py [Tensor] make ColoTensor more robust for getattr (#886) 2022-04-27 10:57:49 +08:00
embedding.py [tensor] derive compute pattern from dist spec (#971) 2022-05-16 14:58:08 +08:00
layernorm.py add DistSpec for loss and test_model (#947) 2022-05-13 20:29:50 +08:00
linear.py [tensor] derive compute pattern from dist spec (#971) 2022-05-16 14:58:08 +08:00
loss.py [tensor] derive compute pattern from dist spec (#971) 2022-05-16 14:58:08 +08:00