From f9de510121be4e9e0f8fd95ed99dd5d877ac8aed Mon Sep 17 00:00:00 2001 From: Jason Ansel Date: Sat, 18 May 2024 17:00:46 -0700 Subject: [PATCH] [dynamo] Graph break on set_num_threads (#126623) Fixes #125364 Pull Request resolved: https://github.com/pytorch/pytorch/pull/126623 Approved by: https://github.com/yanboliang --- .../TestNNDeviceTypeCPU.test_conv_empty_input_cpu_complex128} | 0 torch/_dynamo/trace_rules.py | 3 --- 2 files changed, 3 deletions(-) rename test/{dynamo_expected_failures/TestSortAndSelectCPU.test_sort_overflow_cpu_int16 => dynamo_skips/TestNNDeviceTypeCPU.test_conv_empty_input_cpu_complex128} (100%) diff --git a/test/dynamo_expected_failures/TestSortAndSelectCPU.test_sort_overflow_cpu_int16 b/test/dynamo_skips/TestNNDeviceTypeCPU.test_conv_empty_input_cpu_complex128 similarity index 100% rename from test/dynamo_expected_failures/TestSortAndSelectCPU.test_sort_overflow_cpu_int16 rename to test/dynamo_skips/TestNNDeviceTypeCPU.test_conv_empty_input_cpu_complex128 diff --git a/torch/_dynamo/trace_rules.py b/torch/_dynamo/trace_rules.py index 8a2c12ee4e84..9ac3db864708 100644 --- a/torch/_dynamo/trace_rules.py +++ b/torch/_dynamo/trace_rules.py @@ -1302,9 +1302,6 @@ torch_c_binding_in_graph_functions = dict.fromkeys( "torch._C.parse_schema", "torch._C.parse_type_comment", "torch._C.read_vitals", - "torch._C.set_flush_denormal", - "torch._C.set_num_interop_threads", - "torch._C.set_num_threads", "torch._C.set_vital", "torch._C.unify_type_list", "torch._C.vitals_enabled",