soulitzer e41a0b33ec Allow Fakified subclass to have different device for inner and outer tensor (#141839) ... Previously if a wrapper tensor subclass is fakified, the inner tensors would end up having the same device as the outer tensor. This PR makes it so that inner and outer tensors can have different devices. See OffloadTensor PR https://github.com/pytorch/pytorch/pull/141840/files#diff-3bc0cf540b694f4ec0a3749f78b047456657a53a5657e495ffb68e5970c5fdaaR1955 for an application. A simpler test has been added in this PR. This is technically bc-breaking because now the callback passed to MetaConverter needs to accept an extra argument, but no one external should be using this anyway? Pull Request resolved: https://github.com/pytorch/pytorch/pull/141839 Approved by: https://github.com/bdhirsh ghstack dependencies: #141166		2024-12-03 00:09:41 +00:00
..
__init__.py
_fake_tensor_utils.py
fake_impls.py	[BE]: Apply PERF401 autofixes from ruff (#140980)	2024-11-20 17:52:07 +00:00
fake_tensor.py	Allow Fakified subclass to have different device for inner and outer tensor (#141839)	2024-12-03 00:09:41 +00:00
fake_utils.py	check fake/real mismatches during real tensor prop (#137747)	2024-11-04 23:39:48 +00:00
functional_tensor.py	[BE] Make maybe_aliasing_or_mutating proper tag (#131990)	2024-11-24 00:12:49 +00:00
meta_utils.py	Allow Fakified subclass to have different device for inner and outer tensor (#141839)	2024-12-03 00:09:41 +00:00
schema_check_mode.py