pytorch/torch/csrc/jit/python
Bert Maher 8dda299d96 Re-apply: [nnc] Support thread level parallelism in fused kernels (#63776)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/63776

I reverted this out of an abundance of caution because some test
failures occurred, but they were all due to precision issues fixed lower in
this stack.  Let's try again.

I've rolled the elimination of the allow-parallelism-in-fusions toggle into
this diff since they're pretty tightly coupled.
ghstack-source-id: 136529847

Test Plan: CI

Reviewed By: huiguoo

Differential Revision: D30484555

fbshipit-source-id: 38fd33520f710585d1130c365a8c60c9ce794a59
2021-08-24 18:56:55 -07:00
..
init.cpp Re-apply: [nnc] Support thread level parallelism in fused kernels (#63776) 2021-08-24 18:56:55 -07:00
init.h
module_python.h [JIT] Add reference semantics to TorchScript classes (#44324) 2021-06-30 14:27:17 -07:00
pybind_utils.cpp irange-ify 11 (#62121) 2021-07-28 13:32:09 -07:00
pybind_utils.h Improve IMethod::getArgumentNames to deal with empty argument names list (#62947) 2021-08-11 16:44:00 -07:00
pybind.h irange-ify 11 (#62121) 2021-07-28 13:32:09 -07:00
python_arg_flatten.cpp irange-ify 11 (#62121) 2021-07-28 13:32:09 -07:00
python_arg_flatten.h irange-ify 11 (#62121) 2021-07-28 13:32:09 -07:00
python_custom_class.cpp Make PyTorch code-base clang-tidy compliant (#56892) 2021-04-28 14:10:25 -07:00
python_custom_class.h
python_dict.cpp [JIT] Add torch._C.ScriptDict (#52659) 2021-05-27 10:25:30 -07:00
python_dict.h [JIT] Add torch._C.ScriptDict (#52659) 2021-05-27 10:25:30 -07:00
python_interpreter.cpp Disable avoid-non-const-global-variables lint check (#62008) 2021-07-22 18:04:40 -07:00
python_ir.cpp Test shape analysis with opinfos (#59814) 2021-08-10 09:47:33 -07:00
python_ir.h Fix InlinedCallStack annotation to account for module calling its own (#61791) 2021-07-26 15:00:57 -07:00
python_ivalue.h Modernize override (#61744) 2021-07-23 23:04:46 -07:00
python_list.cpp irange-ify 11 (#62121) 2021-07-28 13:32:09 -07:00
python_list.h [JIT] Add torch._C.ScriptList` (#52832) 2021-07-01 20:28:13 -07:00
python_sugared_value.cpp Support default values on NamedTuple fields (#54682) 2021-06-26 15:18:21 -07:00
python_sugared_value.h Make PyTorch code-base clang-tidy compliant (#56892) 2021-04-28 14:10:25 -07:00
python_tracer.cpp irange-ify 11 (#62121) 2021-07-28 13:32:09 -07:00
python_tracer.h [Usability] Capture argument names for traced functions and modules (#51775) 2021-02-10 18:28:08 -08:00
python_tree_views.cpp Make PyTorch code-base clang-tidy compliant (#56892) 2021-04-28 14:10:25 -07:00
python_tree_views.h
script_init.cpp irange-ify 11 (#62121) 2021-07-28 13:32:09 -07:00
script_init.h
update_graph_executor_opt.cpp Disable avoid-non-const-global-variables lint check (#62008) 2021-07-22 18:04:40 -07:00
update_graph_executor_opt.h