pytorch

86063b4d - Add torch._print to dynamo trace_rules (#120533)

Add torch._print to dynamo trace_rules (#120533) Fixes #114831 Before: ``` (pytorch10) angelayi@devgpu022 ~/local/pytorch [main] $ python test/dynamo/test_trace_rules.py -k test_torch_name_rule_map_updated F ====================================================================== FAIL: test_torch_name_rule_map_updated (__main__.TraceRuleTests) ---------------------------------------------------------------------- Traceback (most recent call last): File "/data/users/angelayi/pytorch/torch/testing/_internal/common_utils.py", line 2739, in wrapper method(*args, **kwargs) File "/data/users/angelayi/pytorch/test/dynamo/test_trace_rules.py", line 328, in test_torch_name_rule_map_updated self._check_set_equality( File "/data/users/angelayi/pytorch/test/dynamo/test_trace_rules.py", line 302, in _check_set_equality self.assertTrue(len(x) == 0, msg1) AssertionError: False is not true : New torch objects: {<built-in method _print of type object at 0x7ff477e40ee0>} were not added to `trace_rules.torch_c_binding_in_graph_functions` or `test_trace_rules.ignored_c_binding_in_graph_function_names`. Refer the instruction in `torch/_dynamo/trace_rules.py` for more details. To execute this test, run the following from the base repo dir: python test/dynamo/test_trace_rules.py -k test_torch_name_rule_map_updated This message can be suppressed by setting PYTORCH_PRINT_REPRO_ON_FAILURE=0 ---------------------------------------------------------------------- Ran 1 test in 0.184s FAILED (failures=1) ``` After change: ``` (pytorch10) angelayi@devgpu022 ~/local/pytorch [main] $ python test/dynamo/test_trace_rules.py -k test_torch_name_rule_map_updated . ---------------------------------------------------------------------- Ran 1 test in 0.209s OK ``` Pull Request resolved: https://github.com/pytorch/pytorch/pull/120533 Approved by: https://github.com/clee2000, https://github.com/yanboliang, https://github.com/huydhn, https://github.com/Skylion007