pytorch

7d220079 - [fx-acc] add acc_op optimization flags and decorator (#65928)

[fx-acc] add acc_op optimization flags and decorator (#65928) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/65928 This diff adds a decorator for adding flags to acc_ops. These flags inform graph optimizations that the op is eligible for optimization by some general criteria (e.g. op acts elementwise, op does quantization). This makes it simpler to expand acc_ops. The user can add an op and add flags to enable optimization without going through all graph opts and trying to determine if new acc_op is eligible for the graph optimization. Even though our list of graph opts is small now we already see that for `sink_reshape_ops` we had hardcoded 11 pointwise acc_ops, now there are 24 pointwise acc_ops. Test Plan: ``` buck test mode/opt glow/fb/fx/graph_opts:test_fx_sink ``` ``` Parsing buck files: finished in 0.5 sec Downloaded 0/3 artifacts, 0.00 bytes, 100.0% cache miss (for updated rules) Building: finished in 37.1 sec (100%) 10279/10279 jobs, 3/10279 updated Total time: 37.7 sec More details at https://www.internalfb.com/intern/buck/build/e13521bb-6142-4960-8cdd-6b5e4780da96 Tpx test run coordinator for Facebook. See https://fburl.com/tpx for details. Running with tpx session id: 16260a2a-d364-4605-9111-6f2a19317036 Trace available for this run at /tmp/tpx-20210922-124332.623880/trace.log Started reporting to test run: https://www.internalfb.com/intern/testinfra/testrun/4222124720425564 ✓ ListingSuccess: glow/fb/fx/graph_opts:test_fx_sink - main (6.038) ✓ Pass: glow/fb/fx/graph_opts:test_fx_sink - test_no_sink_concat_below_quantize (glow.fb.fx.graph_opts.tests.test_fx_sink.TestSink) (0.036) ✓ Pass: glow/fb/fx/graph_opts:test_fx_sink - test_sink_concat_below_quantize (glow.fb.fx.graph_opts.tests.test_fx_sink.TestSink) (0.048) ✓ Pass: glow/fb/fx/graph_opts:test_fx_sink - test_sink_reshape_nodes (glow.fb.fx.graph_opts.tests.test_fx_sink.TestSink) (0.058) ✓ Pass: glow/fb/fx/graph_opts:test_fx_sink - test_no_sink (glow.fb.fx.graph_opts.tests.test_fx_sink.TestSink) (0.057) Summary Pass: 4 ListingSuccess: 1 If you need help understanding your runs, please follow the wiki: https://fburl.com/posting_in_tpx_users Finished test run: https://www.internalfb.com/intern/testinfra/testrun/4222124720425564 ``` Reviewed By: jfix71 Differential Revision: D31121321 fbshipit-source-id: 6f6e3b8e2d57ea30766fa6bee34ca207cec86f0f