rzou 27ec3921bc Optimize mutable torch.library.custom_op overhead (#139513) ... We don't need to do a loop over all the args, kwargs in the AdInplaceOrView key; we just need to bump the version on the args, kwargs that are mutable. On the benchmark mentioned in https://github.com/pytorch/pytorch/issues/139494 this made the time go from ``` mutate2 = 61.72943878173828 no_mutate2 = 36.89440155029297 mutate = 236.3092498779297 no_mutate = 59.31964874267578 ``` to ``` mutate2 = 47.976478576660156 no_mutate2 = 38.37468719482422 mutate = 71.21315002441406 no_mutate = 59.7432975769043 ``` Test Plan: - existing tests Pull Request resolved: https://github.com/pytorch/pytorch/pull/139513 Approved by: https://github.com/bdhirsh ghstack dependencies: #139509		2024-11-05 18:30:53 +00:00
..
__init__.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
autograd.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
custom_ops.py	Optimize mutable torch.library.custom_op overhead (#139513)	2024-11-05 18:30:53 +00:00
fake_class_registry.py	Propagate real tensor tracing with torchbind + fixing side effects (#138797)	2024-10-28 06:27:36 +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]: Update mypy to 1.11.2 (#133816)	2024-09-16 19:44:11 +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	[triton_op] Skip HOP dispatch when possible (#132822)	2024-08-08 15:56:40 +00:00
utils.py	Optimize mutable torch.library.custom_op overhead (#139513)	2024-11-05 18:30:53 +00:00