ColossalAI/colossalai/fx/profiler
Boyuan Yao 132b4306b7
[fx] Add concrete info prop (#1677)
* [fx] concreteinfoprop

* [fx] add concreteinfoprop

* [fx] modify docstring of ConcreteInfoProp

* [fx] fix device error

* [fx] modify parameter calculation

* [fx] modify parameters calculation
2022-10-04 16:48:24 +08:00
..
experimental [fx/profiler] tuned the calculation of memory estimation (#1619) 2022-09-23 10:59:47 +08:00
__init__.py [fx/tuning] tune performance on rotor with meta info. (#1599) 2022-09-15 14:46:36 +08:00
constant.py [fx] refactor code for profiler / enable fake tensor movement. (#1646) 2022-09-27 10:26:52 +08:00
dataflow.py [fx] Add concrete info prop (#1677) 2022-10-04 16:48:24 +08:00
memory.py [fx/profiler] tuned the calculation of memory estimation (#1619) 2022-09-23 10:59:47 +08:00
opcount.py [fx/profiler] tuned the calculation of memory estimation (#1619) 2022-09-23 10:59:47 +08:00
profiler.py [fx] Add concrete info prop (#1677) 2022-10-04 16:48:24 +08:00
tensor.py [fx] refactor code for profiler / enable fake tensor movement. (#1646) 2022-09-27 10:26:52 +08:00