pytorch/torch/csrc/jit/python
David Berard bcc9084bc4 [JIT] Support scripting torch.is_autocast_enabled() (#81305)
This adds an `aten::is_autocast_enabled` op into the jit runtime so that
autocasting ops can be scripted and called from within jit.

Differential Revision: [D37901585](https://our.internmc.facebook.com/intern/diff/D37901585)
Pull Request resolved: https://github.com/pytorch/pytorch/pull/81305
Approved by: https://github.com/qihqi, https://github.com/eellison
2022-07-27 22:32:08 +00:00
..
init.cpp [JIT] Fix python binding error with empty containers in init.cpp (#81786) 2022-07-23 05:50:39 +00: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 Add support for Generator conversion to/from IValue (#81697) 2022-07-19 16:50:10 +00:00
pybind_utils.h Add support for Generator conversion to/from IValue (#81697) 2022-07-19 16:50:10 +00:00
pybind.h [AutoAccept][Codemod][FBSourceClangFormatLinter] Daily arc lint --take CLANGFORMAT 2021-12-19 16:38:25 -08:00
python_arg_flatten.cpp [ONNX] Support optional type (#68793) (#73284) 2022-05-04 20:24:30 +00:00
python_arg_flatten.h irange-ify 11 (#62121) 2021-07-28 13:32:09 -07:00
python_custom_class.cpp
python_custom_class.h
python_dict.cpp
python_dict.h
python_interpreter.cpp [jit] Make operation call accept Stack& instead Stack* (#63414) 2021-08-30 11:49:20 -07:00
python_ir.cpp [JIT] Support scripting torch.is_autocast_enabled() (#81305) 2022-07-27 22:32:08 +00: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 [BE] Fix pybind deprecation warnings (#72376) 2022-02-07 18:33:32 +00:00
python_list.h Fix sign-compare violations in python_list.h 2022-04-01 19:15:51 +00:00
python_sugared_value.cpp turn on -Werror=unused-variable in our Bazel CPU build 2022-06-11 02:46:34 +00:00
python_sugared_value.h [JIT] Support for ParameterDict getattr 2022-05-13 01:03:25 +00:00
python_tracer.cpp Fix C API to be compatible with latest 3.11 beta (#81242) 2022-07-27 08:37:10 +00:00
python_tracer.h
python_tree_views.cpp Reland "Make debug_pkl smaller by only emitting unique traces." (#73368) 2022-04-18 22:34:21 +00:00
python_tree_views.h
script_init.cpp Expose _export_data from C++ to Python (#79207) 2022-06-10 00:41:33 +00: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 Remove WindowsTorchApiMacro.h in favor of Export.h (#69585) 2021-12-09 17:30:09 -08:00