digger yu
|
70c8cdecf4
|
[nfc] fix typo colossalai/cli fx kernel (#3847)
* fix typo colossalai/autochunk auto_parallel amp
* fix typo colossalai/auto_parallel nn utils etc.
* fix typo colossalai/auto_parallel autochunk fx/passes etc.
* fix typo docs/
* change placememt_policy to placement_policy in docs/ and examples/
* fix typo colossalai/ applications/
* fix typo colossalai/cli fx kernel
|
2023-06-02 15:02:45 +08:00 |
|
Super Daniel
|
c41e59e5ad
|
[fx] allow native ckpt trace and codegen. (#2438)
|
2023-01-11 13:49:59 +08:00 |
|
Zihao
|
3a02b46447
|
[auto-parallel] refactoring ColoTracer (#2118)
* add meta_data_computing
* add checkpoint_annotation
* rename proxy.data to proxy.meta_data and add bias addition pass
* polish code
* delete meta_prop_pass invoke and rename ori_node to orig_node
* add TracerType
* unify meta data computing
* delete TracerType
* handle setitem operation
* operator.setitem
|
2023-01-04 14:44:22 +08:00 |
|
Super Daniel
|
2bf2d1cd3b
|
[fx] An experimental version of ColoTracer.' (#2002)
* [fx] add a symbolic_trace api.
* [fx] fix import errors.
* [fx] ColoTracer experimental.
|
2022-12-07 18:36:17 +08:00 |