[pallas backend] Fix load and reduction codegen for mean operations (#…

…170464)

Pull Request resolved: #170464
Approved by: https://github.com/yf225
ghstack dependencies: #170451

[BE][MPS] Don't pass `nnz` to `mark_segments` (#170403)

Fixes following unused variable warning
```
/Users/malfet/git/pytorch/pytorch/aten/src/ATen/native/sparse/mps/kernels/SparseTensorMath.metal:288:27: warning: unused parameter 'nnz' [-Wunused-parameter]
    constant uint&        nnz     [[buffer(2)]],
```

Also, use short circuit language rule to make kernel more compact
Pull Request resolved: #170403
Approved by: https://github.com/Skylion007

Add supported StableIValue types to docs (#168385)

Pull Request resolved: #168385
Approved by: https://github.com/albanD

[c10d] Add thread safety when calling ncclCommGetAsyncError (#170424)

Fixes #169484

Pull Request resolved: #170424
Approved by: https://github.com/kwen2501

[Flex flash]Fix for latest commits (#170481)

Pull Request resolved: #170481
Approved by: https://github.com/v0i0

Revert "[DebugMode] add DebugInterpreter for fx.Node metadata in eage…

…r/aot_eager compiled regions (#170125)"

This reverts commit cf935bc.

Reverted #170125 on behalf of https://github.com/pytorch-auto-revert due to Reverted automatically by pytorch's autorevert, to avoid this behaviour add the tag autorevert: disable ([comment](#170125 (comment)))

Add input_size validation for RNN modules with clear error messages (#…

…166302)

Validates that input_size is an int > 0, providing clear TypeError/ValueError instead of cryptic torch.empty() errors. Applies to nn.RNN, nn.LSTM, nn.GRU.

Fixes #136936

Pull Request resolved: #166302
Approved by: https://github.com/mikaylagawarecki

[CI] fix test_pointwise_ops.py test_mul_div_scalar_partial (#170510)

Support any world size; 2, 3 or 4.

Pull Request resolved: #170510
Approved by: https://github.com/jeffdaily

Co-authored-by: Jeff Daily <[email protected]>

Fix vllm issue for flex (#170499)

Fixess #170499

Pull Request resolved: #170499
Approved by: https://github.com/zou3519

Revert "[DebugMode] add DebugInterpreter for fx.Node metadata in eage…

…r/aot_eager compiled regions (#170125)"

This reverts commit cf935bc.

Reverted #170125 on behalf of https://github.com/pytorch-auto-revert due to Reverted automatically by pytorch's autorevert, to avoid this behaviour add the tag autorevert: disable ([comment](#170125 (comment)))