ColossalAI/colossalai/fx/profiler
Boyuan Yao 90a9fdd91d
[autoparallel] Patch meta information of `torch.matmul` (#2584)
* [autoparallel] matmul metainfo

* [auto_parallel] remove unused print

* [tests] skip test_matmul_handler when torch version is lower than 1.12.0
2023-02-08 11:05:31 +08:00
..
experimental
__init__.py
constants.py
dataflow.py
memory_utils.py
opcount.py [autoparallel] Patch meta information of `torch.matmul` (#2584) 2023-02-08 11:05:31 +08:00
profiler.py
shard_utils.py [autoparallel] modify comm nodes' memory cost in construct chain (#2263) 2023-01-03 11:38:48 +08:00
tensor.py [hotfix] meta tensor default device. (#2510) 2023-01-29 16:28:10 +08:00