pytorch/torch/_subclasses
Tugsbayasgalan (Tugsuu) Manlaibaatar 9409274bc1 Fix bug in functional tensor decomp (#136600)
Summary: Previously we had a very bad bug where we don't allow any decomp on CIA. This never mattered before because we never had to actually push CIA decomp to Python key level in export.

Test Plan: CI

Differential Revision: D63363749

Pull Request resolved: https://github.com/pytorch/pytorch/pull/136600
Approved by: https://github.com/bdhirsh
2024-09-25 17:37:50 +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
_fake_tensor_utils.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
fake_impls.py NJT <-> padded dense conversions (#125947) 2024-09-12 17:54:25 +00:00
fake_tensor.py Make avoid_device_init() aware of hpu device (#136194) 2024-09-24 14:13:45 +00:00
fake_utils.py Move attention kernels back from fake_impls to meta_registrations (#134288) 2024-08-27 21:10:36 +00:00
functional_tensor.py Fix bug in functional tensor decomp (#136600) 2024-09-25 17:37:50 +00:00
meta_utils.py Reland "Construct NJT without graph breaks" (#133196) 2024-08-14 01:11:13 +00:00
schema_check_mode.py Add None return type to init (#132335) 2024-08-01 15:26:45 +00:00