Files
ColossalAI/tests/test_auto_parallel/test_tensor_shard
Boyuan Yao 40c916b192 [autoparallel] Patch meta information of torch.nn.functional.softmax and torch.nn.Softmax (#2674)
* [autoparallel] softmax metainfo

* [autoparallel] softmax metainfo
2023-02-13 16:09:22 +08:00
..