pytorch

mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00

Files

Raman-RH 649ceda8a5 [export] handling NamedTuple inputs (#162959 )

Fixes #160547
### Summary:
bug
```
    def test_namedtuple(self):
        from collections import namedtuple
        Point = namedtuple('Point', 'x y')

        class M(torch.nn.Module):
            def forward(self, x, y):
                return x + y

        inp = Point(torch.ones(3), torch.ones(3))
        print(M()(*inp))

        # errors
        ep = torch.export.export(M(), inp, strict=False)
        print(ep)

        # succeeds
        ep = torch.export.export(M(), inp, strict=True)
        print(ep)

        # workaround could be to convert namedtuple to a kwarg
        inp_kwargs =  {field: getattr(inp, field) for field in inp._fields}
        ep = torch.export.export(M(), (), inp_kwargs)
        print(ep)
```
FIx :
namedtuple is subclass of tuple
but namedtuple is not expected
So, this change handles named tuple case

I have added 🧪 test case for this as well
Pull Request resolved: https://github.com/pytorch/pytorch/pull/162959
Approved by: https://github.com/angelayi

Co-authored-by: Angela Yi <angelayi@meta.com>

2025-09-23 17:43:50 +00:00

experimental

[Torch Native] Add test for packaging weight (#158750 )

2025-08-09 01:04:21 +00:00

passes

[export] Move example inputs in move_to_device_pass (#162301 )

2025-09-06 23:54:54 +00:00

pt2_archive

[export] Remove .contiguous() when saving weights to raw bytes (#163587 )

2025-09-23 15:44:56 +00:00

__init__.py

Fix inconsistent test and add new tracer as config (#162558 )