pytorch

pytorch copied to clipboard

[Reland] Verify types in custom op schemas

4

Summary: co-dev reland of https://github.com/pytorch/pytorch/pull/124520, which requires the removal of some executorch tests. Before this PR, we didn't check that types in a schema were valid. This is because TorchScript...

zou3519

[inductor] make conv lowering work with dynamic shapes

3

Stack from [ghstack](https://github.com/ezyang/ghstack) (oldest at bottom): * __->__ #126823 Fix an issue reported by internal user that conv lowering does not work well with dynamic shapes. cc @voznesenskym @penguinwu @EikanWang...

shunting314

Turn on fakification by deafult in non-strict export

2

Summary: To check what fake class we need to add to make all tests pass. Test Plan: Existing tests. Differential Revision: D57678693

ydwu4

`@functools.wraps` graph breaks in many cases where we should be able to handle it

1

### 🐛 Describe the bug Currently, PyTorch crashes if two functions annotated with `@functools.wraps(func)` are composed together. As an example, when adding support for `torch.func.hessian` in https://github.com/pytorch/pytorch/pull/121410, `wraps` is conditionally...

guilhermeleobas

Change error message when nn module inlining is enabled for MiscTests.test_map_side_effects

5

#fix https://github.com/pytorch/pytorch/issues/126355 Stack from [ghstack](https://github.com/ezyang/ghstack) (oldest at bottom): * #126904 * #126693 * __->__ #126444 cc @voznesenskym @penguinwu @EikanWang @jgong5 @Guobing-Chen @XiaobingSuper @zhuhaozhe @blzheng @wenzhe-nrv @jiayisunx @chenyang78 @kadeng @chauhang

laithsakka

[Inductor][CPP] Add Min/Max with VecMask

1

Stack from [ghstack](https://github.com/ezyang/ghstack) (oldest at bottom): * __->__ #126841 **Summary** Fix issue: https://github.com/pytorch/pytorch/issues/126824 which is missing the support of `min/max` with `VecMask`. **TestPlan** ``` python test/inductor/test_torchinductor_opinfo.py -k test_comprehensive_clamp_max_cpu_bool python test/inductor/test_torchinductor_opinfo.py...

leslie-fang-intel

[Docs] Fix NumPy + backward example

5

Stack from [ghstack](https://github.com/ezyang/ghstack) (oldest at bottom): * __->__ #126872 We were calling backward on a tensor not a scalar...

lezcano

Fix nextafter in inductor CPP codegen

1

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

isuruf

[OUTDATED!, Autograd] Cond Higher-Order Operation

1

This PR is intended to provide the Autograd mechanism for the `cond` operation. @ydwu4

bohnstingl