pytorch

pytorch copied to clipboard

[Inductor] Make FlexAttention block_mask argument as tuple

1

Stack from [ghstack](https://github.com/ezyang/ghstack) (oldest at bottom): * __->__ #129831 Re-organize ```block_mask``` related arguments a tuple to reduce the individual argument number. I was trying to use named tuple, but aot...

yanboliang

[inductor] switch CppCodeCache to new cpp_builder.

25

Original PRs is damaged by confilct and rebase: https://github.com/pytorch/pytorch/pull/128303, https://github.com/pytorch/pytorch/pull/129144 This PR just switch `CppCodeCache` to new cpp_builder, it need to be validated in `fb_code` environment. cc @peterjc123 @mszhanyi @skyline75489...

xuhancn

[dynamo] Skip FUNCTION_MATCH on method-wrapper objects

1

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

anijain2305

Bug fix for captuing execution trace grid function

2

Summary: The inputs to grid function are varying argument, it can be one number, two numbers, or three numbers. The current implementation captured it as a tuple. For example "grid((16,))"....

shengfukevin

[2/N] Fix clang-tidy warnings in torch/csrc/jit/serialization

12

Follows #129055

cyyever

Make `_embedding_bag_backward` explicitly dispatch to CPU and CUDA.

3

Stack from [ghstack](https://github.com/ezyang/ghstack) (oldest at bottom): * __->__ #129691 This PR modifies `_embedding_bag_backward` item inside _native_functions.yaml_, so that it dispatches to CPU and CUDA directly, instead of `CompositeImplicitAutograd`. *Context:* PyTorch...

ysiraichi

[Ez][BE]: Enable new stable ruff rules

7

Applies a bunch of new ruff lint rules that are now stable. Some of these improve efficiency or readability. Since I already did passes on the codebase for these when...

Skylion007

[BE][Easy][13/19] enforce style for empty lines in import segments in `test/j*/`

1

See https://github.com/pytorch/pytorch/pull/129751#issue-2380881501. Most changes are auto-generated by linter. You can review these PRs via: ```bash git diff --ignore-all-space --ignore-blank-lines HEAD~1 ``` Stack from [ghstack](https://github.com/ezyang/ghstack) (oldest at bottom): * __->__ #129764...

XuehaiPan

smart flushing

9

Fixes #ISSUE_NUMBER cc @voznesenskym @penguinwu @EikanWang @jgong5 @Guobing-Chen @XiaobingSuper @zhuhaozhe @blzheng @wenzhe-nrv @jiayisunx @peterbell10 @ipiszy @yf225 @chenyang78 @kadeng @muchulee8 @ColinPeppler @amjames @desertfire @chauhang

nmacchioni

[dynamo] Validate check_fn

13

Stack from [ghstack](https://github.com/ezyang/ghstack) (oldest at bottom): * __->__ #118448 * #129830 cc @voznesenskym @penguinwu @EikanWang @jgong5 @Guobing-Chen @XiaobingSuper @zhuhaozhe @blzheng @wenzhe-nrv @jiayisunx @chenyang78 @kadeng @chauhang @aakhundov

anijain2305