IvanKobzarev a37afd23fa [custom_ops][perf] Move expensive pytree traversals of tensors to C++ (#148555) ... (benchmark for 1 call) Before: ``` └─ $ python ~/task_custom_ops_perf/test_custom_ops_perf_repro.py DO_BENCH mutate: 77.72445678710938 us PROFILE:/home/ivankobzarev/task_custom_ops_perf/mutate.json DO_BENCH no_mutate: 64.61143493652344 us PROFILE:/home/ivankobzarev/task_custom_ops_perf/no_mutate.json DO_BENCH direct_mutate: 11.682510375976562 us PROFILE:/home/ivankobzarev/task_custom_ops_perf/direct_mutate.json DO_BENCH direct_no_mutate: 18.596649169921875 us PROFILE:/home/ivankobzarev/task_custom_ops_perf/direct_no_mutate.json ``` After: ``` └─ $ python ~/task_custom_ops_perf/test_custom_ops_perf_repro.py DO_BENCH mutate: 47.6837158203125 us PROFILE:/home/ivankobzarev/task_custom_ops_perf/mutate.json DO_BENCH no_mutate: 31.709671020507812 us PROFILE:/home/ivankobzarev/task_custom_ops_perf/no_mutate.json DO_BENCH direct_mutate: 10.967254638671875 us PROFILE:/home/ivankobzarev/task_custom_ops_perf/direct_mutate.json DO_BENCH direct_no_mutate: 10.728836059570312 us PROFILE:/home/ivankobzarev/task_custom_ops_perf/direct_no_mutate.json ``` Pull Request resolved: https://github.com/pytorch/pytorch/pull/148555 Approved by: https://github.com/zou3519		2025-04-01 18:45:48 +00:00
..
__init__.py	fix typo in doc and import for torch._library.triton (#144882)	2025-01-17 17:32:12 +00:00
autograd.py	[custom_ops][perf] Move expensive pytree traversals of tensors to C++ (#148555)	2025-04-01 18:45:48 +00:00
custom_ops.py	[custom_ops][perf] Move expensive pytree traversals of tensors to C++ (#148555)	2025-04-01 18:45:48 +00:00
fake_class_registry.py	[torchbind] Differentiate ScriptModule and ScriptObject with qualified name (#147399)	2025-02-20 04:57:57 +00:00
fake_impl.py	[real_tensor_prop] Infer Fake kernels during real tensor prop (#139213)	2024-10-30 21:29:33 +00:00
infer_schema.py	[BE][Ez]: ISC001 Auto concatenate implicit one line strings (#146408)	2025-02-04 19:07:04 +00:00
simple_registry.py	[BE][Easy][14/19] enforce style for empty lines in import segments in torch/_[a-c]*/ and torch/_[e-h]*/ and torch/_[j-z]*/ (#129765)	2024-07-31 10:42:50 +00:00
triton.py	Typo Errors fixed in multiple files (#148262)	2025-03-09 12:21:40 +00:00
utils.py	[custom_ops][perf] Move expensive pytree traversals of tensors to C++ (#148555)	2025-04-01 18:45:48 +00:00