pytorch

pytorch copied to clipboard

[dtensor][4/N] refactor dispatching logic and add propagator

1

Stack from [ghstack](https://github.com/ezyang/ghstack) (oldest at bottom): * #92126 * #90735 * #90734 * __->__ #90733 * #92931 * #92930 * #93040 This PR refactors the dispatching logic to make it...

wanchaol

Properly compute device for elementwise operations with CPU scalar tensor

4

Stack from [ghstack](https://github.com/ezyang/ghstack) (oldest at bottom): * __->__ #93073 Signed-off-by: Edward Z. Yang

ezyang

More tidy fixes

1

cc @jgong5 @mingfeima @XiaobingSuper @sanchitintel @ashokei @jingxu10

cyyever

add numpy typing plugin to mypy config

9

Stack from [ghstack](https://github.com/ezyang/ghstack) (oldest at bottom): * #92126 * #90735 * #90734 * #90733 * #92931 * __->__ #92930 * #93040 This added the numpy typing plugin to mypy config...

wanchaol

[jit] jit.ignore funcs ignore non-jit return type annotations

5

`@torch.jit.unused` is supposed to mark methods to be excluded from jit scripting. jit creates stub with the same number of arguments to throw an exception if client call `jit.unused` method....

IvanKobzarev

add torch.autograd._set_view_replay_enabled, use in aot autograd

6

tldr; this should fix some minor perf regressions that were caused by adding more as_strided() calls in aot autograd. This PR adds a new context manager, `torch.autograd._set_view_replay_enabled()`. Context: AOT Autograd...

bdhirsh

Make CPU inductor work with dynamic shapes

1

Stack from [ghstack](https://github.com/ezyang/ghstack) (oldest at bottom): * __->__ #93077 These errors were found by looking at wav2vec2 See https://github.com/pytorch/pytorch/issues/91719 Signed-off-by: Edward Z. Yang cc @mlazos @soumith @voznesenskym @yanboliang @penguinwu @anijain2305...

ezyang

Get rid of dedicated inductor dynamic_shapes config

1

Stack from [ghstack](https://github.com/ezyang/ghstack) (oldest at bottom): * __->__ #93076 Instead, use Dynamo dynamic_shapes config Signed-off-by: Edward Z. Yang cc @mlazos @soumith @voznesenskym @yanboliang @penguinwu @anijain2305 @EikanWang @jgong5 @Guobing-Chen @chunyuan-w @XiaobingSuper...

ezyang

[dtensor][7/N] remove backend in with_comms

1

Stack from [ghstack](https://github.com/ezyang/ghstack) (oldest at bottom): * #92126 * #90735 * #90734 * #90733 * #92931 * #92930 * __->__ #93040 backend is not actually getting used in anywhere, so...

wanchaol

Make meshgrid support symbolic shapes

1

Stack from [ghstack](https://github.com/ezyang/ghstack) (oldest at bottom): * __->__ #93075 Signed-off-by: Edward Z. Yang cc @gujinghui @PenghuiCheng @XiaobingSuper @jianyuh @jgong5 @mingfeima @sanchitintel @ashokei @jingxu10 @min-jean-cho @yanbing-j @Guobing-Chen @Xia-Weiwen

ezyang