pytorch/torch/csrc/jit/python
Richard Zou f8857cef45 [Reland] Verify types in custom op schemas (#126861)
Summary:
co-dev reland of https://github.com/pytorch/pytorch/pull/124520, which requires
the removal of some executorch tests.

Before this PR, we didn't check that types in a schema were valid. This
is because TorchScript treats unknown types as type variables.

This PR checks types in a schema for the TORCH_LIBRARY APIs. To do this,
we add an `allow_typevars` flag to parseSchema so that TorchScript can
use allow_typevars=True. We also add some error messages for common
mistakes (e.g. using int64_t or double in schema).

Test Plan: Wait for tests

Differential Revision: D57666659

Pull Request resolved: https://github.com/pytorch/pytorch/pull/126861
Approved by: https://github.com/albanD
2024-05-23 19:53:52 +00:00
..
init.cpp [Reland] Verify types in custom op schemas (#126861) 2024-05-23 19:53:52 +00:00
init.h
module_python.h [codemod] c10:optional -> std::optional (#126135) 2024-05-14 19:35:51 +00:00
pybind_utils.cpp [codemod] c10:optional -> std::optional (#126135) 2024-05-14 19:35:51 +00:00
pybind_utils.h [codemod] c10:optional -> std::optional (#126135) 2024-05-14 19:35:51 +00:00
pybind.h
python_arg_flatten.cpp
python_arg_flatten.h
python_custom_class.cpp
python_custom_class.h
python_dict.cpp
python_dict.h
python_interpreter.cpp
python_ir.cpp [codemod] c10:optional -> std::optional (#126135) 2024-05-14 19:35:51 +00:00
python_ir.h [codemod] c10:optional -> std::optional (#126135) 2024-05-14 19:35:51 +00:00
python_ivalue.h [codemod] c10:optional -> std::optional (#126135) 2024-05-14 19:35:51 +00:00
python_list.cpp
python_list.h [codemod] c10:optional -> std::optional (#126135) 2024-05-14 19:35:51 +00:00
python_sugared_value.cpp [codemod] c10:optional -> std::optional (#126135) 2024-05-14 19:35:51 +00:00
python_sugared_value.h [codemod] c10:optional -> std::optional (#126135) 2024-05-14 19:35:51 +00:00
python_tracer.cpp [codemod] c10:optional -> std::optional (#126135) 2024-05-14 19:35:51 +00:00
python_tracer.h
python_tree_views.cpp [codemod] c10:optional -> std::optional (#126135) 2024-05-14 19:35:51 +00:00
python_tree_views.h
script_init.cpp Use object identity for deepcopy memo (#126126) 2024-05-17 00:06:26 +00:00
script_init.h
update_graph_executor_opt.cpp
update_graph_executor_opt.h
utf8_decoding_ignore.cpp
utf8_decoding_ignore.h