frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 12:54:11 +08:00
Code Issues Packages Projects Releases Wiki Activity

[dynamo] Graph break on set_num_threads (#126623)

Browse Source 
Fixes #125364

Pull Request resolved: https://github.com/pytorch/pytorch/pull/126623
Approved by: https://github.com/yanboliang
This commit is contained in:
Jason Ansel
2024-05-18 17:00:46 -07:00
committed by PyTorch MergeBot
parent 89c1cfe144
commit f9de510121
2 changed files with 0 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

0
test/dynamo_expected_failures/TestSortAndSelectCPU.test_sort_overflow_cpu_int16 → test/dynamo_skips/TestNNDeviceTypeCPU.test_conv_empty_input_cpu_complex128
View File

3
torch/_dynamo/trace_rules.py
View File

@ -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",