mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 12:21:27 +01:00
1803a592f4
Summary: Add script to go through view ops in "native_functions.yaml" and auto-register them into static runtime and auto-generate op unit tests for each. Overall there are 96 grouped view ops, among which 21 is already registered by hand; 9 (including sparse ops/training related ops etc.) are not the target of static runtime; 30 has list args or list ret; and 7 has non-basic types such as "Dimname", "MemoryFormat", etc. In summary, this script auto-generate 29 view ops for now. Run `buck run //caffe2/torch/fb/jit:gen_static_runtime_ops` to generate static runtime ops, and the results with this script are, ``` total grouped native ops: 1582 grouped native ops with out variant: 548 generated functions groups with out variant: 241 view grouped native ops: 96 generated functions view groups: 29 overall generated : 270 ``` The generated view ops are added in D36258968 Test Plan: Generate static runtime ops: `buck run //caffe2/torch/fb/jit:gen_static_runtime_ops` Unit tests: `buck run mode/opt //caffe2/benchmarks/static_runtime:static_runtime_cpptest` Differential Revision: D36258767 Pull Request resolved: https://github.com/pytorch/pytorch/pull/77105 Approved by: https://github.com/mikeiovine |
||
|---|---|---|
| .. | ||
| __init__.py | ||
| config.py | ||
| gen_static_runtime_ops.py | ||
| generator.py | ||