[1.5 cherrypick] Fix torch.mm in ONNX exporter

[houseroad]

Original PR: #34661

[dr-ci]

💊 CircleCI build failures summary and remediations

As of commit f28d51c (more details on the Dr. CI page):

---

• 3/3 failures introduced in this PR

---

🕵️ 2 new failures recognized by patterns

The following build failures do not appear to be due to upstream breakages (reran 2 jobs to discount flakiness):

pytorch_linux_xenial_py3_6_gcc5_4_test (1/2)

Step: "Test" (full log | pattern match details) <confirmed not flaky by 2 failures>

Mar 25 22:05:09 AssertionError: 11 not less than or equal to 1e-05 :

Mar 25 22:05:09 ---------------------------------------------------------------------- 
Mar 25 22:05:09 Traceback (most recent call last): 
Mar 25 22:05:09   File "/opt/conda/lib/python3.6/site-packages/torch/testing/_internal/common_distributed.py", line 175, in wrapper 
Mar 25 22:05:09     self._join_processes(fn) 
Mar 25 22:05:09   File "/opt/conda/lib/python3.6/site-packages/torch/testing/_internal/common_distributed.py", line 285, in _join_processes 
Mar 25 22:05:09     self._check_return_codes(elapsed_time) 
Mar 25 22:05:09   File "/opt/conda/lib/python3.6/site-packages/torch/testing/_internal/common_distributed.py", line 328, in _check_return_codes 
Mar 25 22:05:09     self.assertEqual(first_process.exitcode, 0) 
Mar 25 22:05:09   File "/opt/conda/lib/python3.6/site-packages/torch/testing/_internal/common_utils.py", line 915, in assertEqual 
Mar 25 22:05:09     super(TestCase, self).assertLessEqual(abs(x - y), prec, message) 
Mar 25 22:05:09 AssertionError: 11 not less than or equal to 1e-05 :  
Mar 25 22:05:09  
Mar 25 22:05:09 ---------------------------------------------------------------------- 
Mar 25 22:05:09 Ran 27 tests in 32.861s 
Mar 25 22:05:09  
Mar 25 22:05:09 FAILED (failures=1) 
Mar 25 22:05:09  
Mar 25 22:05:09 Generating XML reports... 
Mar 25 22:05:09 Traceback (most recent call last): 
Mar 25 22:05:09   File "test/run_test.py", line 674, in <module> 
Mar 25 22:05:09     main() 

pytorch_linux_xenial_cuda10_2_cudnn7_py3_gcc7_test (2/2)

Step: "Test" (full log | pattern match details) <confirmed not flaky by 2 failures>

Mar 25 22:40:55 AssertionError: 11 not less than or equal to 1e-05 :

Mar 25 22:40:55 ---------------------------------------------------------------------- 
Mar 25 22:40:55 Traceback (most recent call last): 
Mar 25 22:40:55   File "/opt/conda/lib/python3.6/site-packages/torch/testing/_internal/common_distributed.py", line 175, in wrapper 
Mar 25 22:40:55     self._join_processes(fn) 
Mar 25 22:40:55   File "/opt/conda/lib/python3.6/site-packages/torch/testing/_internal/common_distributed.py", line 285, in _join_processes 
Mar 25 22:40:55     self._check_return_codes(elapsed_time) 
Mar 25 22:40:55   File "/opt/conda/lib/python3.6/site-packages/torch/testing/_internal/common_distributed.py", line 328, in _check_return_codes 
Mar 25 22:40:55     self.assertEqual(first_process.exitcode, 0) 
Mar 25 22:40:55   File "/opt/conda/lib/python3.6/site-packages/torch/testing/_internal/common_utils.py", line 915, in assertEqual 
Mar 25 22:40:55     super(TestCase, self).assertLessEqual(abs(x - y), prec, message) 
Mar 25 22:40:55 AssertionError: 11 not less than or equal to 1e-05 :  
Mar 25 22:40:55  
Mar 25 22:40:55 ---------------------------------------------------------------------- 
Mar 25 22:40:55 Ran 27 tests in 22.071s 
Mar 25 22:40:55  
Mar 25 22:40:55 FAILED (failures=1) 
Mar 25 22:40:55  
Mar 25 22:40:55 Generating XML reports... 
Mar 25 22:40:55 Traceback (most recent call last): 
Mar 25 22:40:55   File "test/run_test.py", line 674, in <module> 
Mar 25 22:40:55     main() 

---

1 job timed out:

• pytorch_macos_10_13_py3_test

---

This comment was automatically generated by Dr. CI (expand for details).

Follow this link to opt-out of these comments for your Pull Requests.

Please report bugs/suggestions on the GitHub issue tracker.

See how this bot performed.

This comment has been revised 5 times.