pytorch

539acb29 - [Static Runtime] Fix a broken test & Add an out variant wrapper for `mse_loss` (#73574)

[Static Runtime] Fix a broken test & Add an out variant wrapper for `mse_loss` (#73574) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/73574 T113070663 identified a test breakage in `StaticRuntime.autogen__convert_indices_from_csr_to_coo` using `mode/opt`. This change fixes it by using a correct test value. By generating the out variants for Static Runtime, this change also includes an out variant wrapper for `mse_loss`. **Generating out variants for Static Runtime** ``` [djang@devbig024.ftw2 ~/fbsource/fbcode/caffe2] buck run //caffe2/torch/fb/jit:gen_static_runtime_ops Invalidating internal cached state: Buck configuration options changed between invocations. This may cause slower builds. Changed value //fbcode.sanitizer='address-undefined-dev' (was 'thread') ... and 13 more. See logs for all changes Parsing buck files: finished in 0.8 sec Downloaded 14/25 artifacts, 159.45 Kbytes, 30.0% cache miss (for updated rules) Building: finished in 1.6 sec (100%) 52/52 jobs, 35/52 updated Total time: 2.5 sec BUILD SUCCEEDED total grouped native ops: 1501 structured grouped native ops: 540 generated grouped native ops: 137 ``` Test Plan: Ran the broken test in `mode/opt` and confirmed that it passes now. ``` [djang@devbig024.ftw2 ~/fbsource/fbcode/caffe2] buck test mode/opt //caffe2/benchmarks/static_runtime:static_runtime_cpptest -- --exact 'caffe2/benchmarks/static_runtime:static_runtime_cpptest - StaticRuntime.autogen__convert_indices_from_csr_to_coo' --run-disabled Invalidating internal cached state: Buck configuration options changed between invocations. This may cause slower builds. Changed value //project.buck_out='buck-out/opt' (was 'buck-out/dev') ... and 307 more. See logs for all changes DEBUG: /data/users/djang/fbsource/tools/build_defs/fbcode_macros/build_defs/lib/cpp_common.bzl:287:14: Using disallowed linker flag 'arvr/third-party/toolchains/platform009/build/mesa/lib/libGL.so' in library rule 'fbsource//third-party/toolchains:opengl' DEBUG: /data/users/djang/fbsource/tools/build_defs/fbcode_macros/build_defs/lib/cpp_common.bzl:287:14: Using disallowed linker flag 'arvr/third-party/freeglut/3.0.0/libs/x64-linux/libglut.a' in library rule 'fbsource//third-party/toolchains:GLUT' I0301 08:28:08.884272 2239319 configeratorc.cpp:70] Attempting to get config buck/detectors/bypass_dirty_builds, timeout=10000 I0301 08:30:14.751745 2261718 configeratorc.cpp:70] Attempting to get config buck/detectors/bypass_dirty_builds, timeout=10000 Parsing buck files: finished in 10.1 sec Creating action graph: finished in 6.1 sec [RE] Metadata: Session ID=[https://fburl.com/b/reSessionID-fa0ba93b-33a1-4e6f-88f8-9f508d2c27c3] [RE] Waiting on 0 remote actions. Completed 247 actions remotely, action cache hit rate: 0.00%. Downloaded 13000/17457 artifacts, 463.99 Mbytes, 2.6% cache miss (for updated rules) Building: finished in 04:16.6 min (100%) 28628/28628 jobs, 28628/28628 updated Total time: 04:32.9 min More details at https://www.internalfb.com/intern/buck/build/c774ff43-5311-49ce-a677-30e3f6afdad1 BUILD SUCCEEDED Tpx test run coordinator for Facebook. See https://fburl.com/tpx for details. Running with tpx session id: 16d9b24c-4a63-4671-84b5-690fac0ee086 Trace available for this run at /tmp/tpx-20220301-083049.472831-16d9b24c-4a63-4671-84b5-690fac0ee086/trace.log RemoteExecution session id: reSessionID-16d9b24c-4a63-4671-84b5-690fac0ee086-tpx Started reporting to test run: https://www.internalfb.com/intern/testinfra/testrun/4503599719295685 ✓ ListingSuccess: caffe2/benchmarks/static_runtime:static_runtime_cpptest : 285 tests discovered (0.425) ✓ Pass: caffe2/benchmarks/static_runtime:static_runtime_cpptest - StaticRuntime.autogen__convert_indices_from_csr_to_coo (0.105) Summary Pass: 1 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/4503599719295685 ``` Reviewed By: mikeiovine Differential Revision: D34552645 fbshipit-source-id: 36f15b0f29edcb7deb71ba8a6f66ce2532bf7c82 (cherry picked from commit 2329afd8bfc89671cfbd864414e528241e7045fc)