pytorch

mirror of https://github.com/zebrajr/pytorch.git synced 2026-01-15 12:15:51 +00:00

Files

pzzp 4375ee9661 Fix AdaptiveMaxPool index error (#159936 )

Fix failure of:
```python
import torch
x = torch.zeros([1, 1, 65536], dtype=torch.float16)
x[0, 0, -1] = 1
_, indices = torch.adaptive_max_pool1d(x, [1])
assert indices == torch.argmax(x.reshape(-1))
```

Pull Request resolved: https://github.com/pytorch/pytorch/pull/159936
Approved by: https://github.com/jerryzh168, https://github.com/cyyever

2025-12-22 17:07:09 +00:00

attention

Add determinsitic flag for fa4 integration (#170887 )

2025-12-20 03:09:27 +00:00

test_convolution.py

Remove outdated CUDA and ROCm skip conditions (#170868 )

2025-12-20 07:06:49 +00:00

test_dropout.py

[MPS] Fix random in-place ops on non-contiguous tensors (#165267 )

2025-10-28 18:43:22 +00:00

test_embedding.py

Fix EmbeddingBag when input is 2D and include_last_offset is True (#168159 )