Angela Yi

Add pixel_shuffle to core aten decomps

4

Summary: https://github.com/pytorch/pytorch/pull/118239 added a decomposition for pixel_shuffle, so pixel_shuffle no longer needs to be a Core ATen Op. We have also fixed the internal use case so that it no...

[export] Change runtime asserts to using assert_scalar

20

By changing runtime symbolic asserts to using assert_scalar, the asserts can call into `expect_true` and modify the shape env so that we can run through the traced graph module with...

Prevent converting runtime-assertion related ops

4

Reviewed By: tarun292 Differential Revision: D56209661

[aoti] Add pt2 package save/load (python)

1

Stack from [ghstack](https://github.com/ezyang/ghstack) (oldest at bottom): * __->__ #125054 cc @voznesenskym @penguinwu @EikanWang @jgong5 @Guobing-Chen @XiaobingSuper @zhuhaozhe @blzheng @wenzhe-nrv @jiayisunx @peterbell10 @ipiszy @yf225 @chenyang78 @kadeng @muchulee8 @ColinPeppler @amjames @desertfire @chauhang

[export] Don't create a new fake mode if dynamo tracing

1

Fixes #ISSUE_NUMBER

[export] Fix for unflattening modules with duplicate tensors

4

In the given test case, we have a ModuleList of 3 modules (`norm.0`, `norm.1`, `norm.2`) which share the same `weight` and `bias` tensors. However when we trace, they all end...

[aoti] Add initial custom op support

1

Stack from [ghstack](https://github.com/ezyang/ghstack) (oldest at bottom): * __->__ #125242 cc @ezyang @msaroufim @bdhirsh @anijain2305 @chauhang @voznesenskym @penguinwu @EikanWang @jgong5 @Guobing-Chen @XiaobingSuper @zhuhaozhe @blzheng @wenzhe-nrv @jiayisunx @peterbell10 @ipiszy @yf225 @chenyang78 @kadeng...

[export] Add metadata for nodes insert_deferred_runtime_asserts

7

Fixes [internal error](https://fb.workplace.com/groups/1075192433118967/permalink/1416709435633930/). The issue is that the asserting nodes added in the `insert_deferred_runtime_assertion` pass do not contain metadata that the ExportedProgram requires the graph to have. One solution to...

[export] Fix serialization of empty torch artifact

1

A previous PR added support for serializing/deserializing example inputs, but this fails when `example_inputs` is none.

[export] Allow constant attr mutation

5

Test Plan: CI Differential Revision: D56893728 cc @voznesenskym @penguinwu @EikanWang @jgong5 @Guobing-Chen @XiaobingSuper @zhuhaozhe @blzheng @wenzhe-nrv @jiayisunx @chenyang78 @kadeng @chauhang