Zihao
|
12e7bcd720
|
register meta func for rnn (#2159)
|
2022-12-21 23:06:18 +08:00 |
|
Zihao
|
a128eec9d5
|
register aten._convolution.default (#2137)
|
2022-12-18 19:27:01 +08:00 |
|
Super Daniel
|
2edbef13cc
|
[fx] add more meta_registry for MetaTensor execution. (#2000)
* [sc] add examples for auto checkpoint.
* merge upstream
* [fx] add more meta_registry for MetaTensor execution.
|
2022-11-23 10:55:46 +08:00 |
|
Super Daniel
|
393f594051
|
[fx/meta/rpc] move _meta_registration.py to fx folder / register fx functions with compatibility checks / remove color debug (#1710)
* [fx] move meta registration
* [fx] fix tests.
* [fx] fix test.
* [fx] fix.
* [meta] refactor meta registration.py.
* [fx] add compatibility descriptions.
* [fx] polish import.
* [fx] add a decorator.
* [fx] fix tests.
* [fx] remove print.
* [fx] edit raise error.
* [fx] edit raise error.
* [fx] add type hint.
* [fx] fix import in experimental.
* [rpc] remove color debug.
* [meta] fix naming.
|
2022-10-18 10:44:23 +08:00 |