Richard Zou ad0883a288 [real_tensor_prop] Infer Fake kernels during real tensor prop (#139213) ... This PR changes real_tensor_prop to also infer fake kernels when the operator doesn't have it. We infer the fake output to be of the same properties as the real output, with unbacked symints in the sizes and some stride order. Test Plan: - new tests Pull Request resolved: https://github.com/pytorch/pytorch/pull/139213 Approved by: https://github.com/pianpwk ghstack dependencies: #139212		2024-10-30 21:29:33 +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	[real tensor prop] Add some asserts for custom ops (#139212)	2024-10-30 19:29:11 +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	[real_tensor_prop] Infer Fake kernels during real tensor prop (#139213)	2024-10-30 21:29:33 +00:00