Sheng Fu 31bfa59970 Capture primitive data type arguments for profiling python_function (#120949) ... RECORD_FUNCTION in python_function only captures argument that is a Tensor. However, it is very common for user to use non tensor arguments in custom ops, for example, sequence length in GPT attention custom op. My previous PR tries to capture all non-tensor arguments, it turned out in some cases, it is very expensive. This PR is to support primitive (or its container) arguments in RECORD_FUNCTION. Pull Request resolved: https://github.com/pytorch/pytorch/pull/120949 Approved by: https://github.com/soulitzer		2024-03-06 05:09:22 +00:00
..
init.cpp	Release the GIL in serialization when it is safe to do so (#120818)	2024-03-01 22:37:26 +00:00
init.h
module_python.h
pybind_utils.cpp	Fix gradient refcounts in pybind and compiled autograd (#118817)	2024-02-07 10:25:42 +00:00
pybind_utils.h	Capture primitive data type arguments for profiling python_function (#120949)	2024-03-06 05:09:22 +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	[2/N] Replace std::tie with structural binding (#119879)	2024-02-15 02:56:34 +00:00
python_ir.h
python_ivalue.h
python_list.cpp
python_list.h
python_sugared_value.cpp	[2/N] Replace std::tie with structural binding (#119879)	2024-02-15 02:56:34 +00:00
python_sugared_value.h
python_tracer.cpp
python_tracer.h
python_tree_views.cpp	[2/N] Replace std::tie with structural binding (#119879)	2024-02-15 02:56:34 +00:00
python_tree_views.h
script_init.cpp	[2/N] Replace std::tie with structural binding (#119879)	2024-02-15 02:56:34 +00:00
script_init.h
update_graph_executor_opt.cpp
update_graph_executor_opt.h
utf8_decoding_ignore.cpp
utf8_decoding_ignore.h