Joel Schlosser 0ebfa3d7d2 Avoid fast path mask left-align check in compiled TransformerEncoder (#163773) ...

Fixes #163640

This PR avoids a mask left align check in the case that we're operating under torch.compile / torch.export. Originally, I planned to make a more invasive change to auto-disable the fast path entirely underneath torch.compile / torch.export, but I realized during testing that the fast path wasn't actually causing compile issues outside of the narrow issue identified here.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/163773
Approved by: https://github.com/mikaylagawarecki

2025-09-26 22:29:37 +00:00

..

test_convolution.py

[cuDNN][conv][64-bit] Disable cuDNN for 64-bit depthwise convs again (#163171)

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

test_dropout.py

Enable additional tests for MPS CI runs (#134356)

2024-10-04 21:52:38 +00:00

test_embedding.py

Fix largeTensorTest malfunction on XPU (#161988)

2025-09-04 16:10:03 +00:00

test_init.py

Fix unused Python variables in test/nn (#143396)

2024-12-18 03:30:54 +00:00

test_lazy_modules.py

Typo fixes for "overridden" in comments and function names (#155944)

2025-06-14 03:37:38 +00:00

test_load_state_dict.py

[BE][PYFMT] migrate PYFMT for {torch,test}/{nn,optim}/** to ruff format (#144548)

2025-06-14 11:27:04 +00:00

test_module_hooks.py

Add warning for module full backward hook when no input requires gradient (#155339)

2025-06-10 04:42:06 +00:00

test_multihead_attention.py

Avoid fast path mask left-align check in compiled TransformerEncoder (#163773)

2025-09-26 22:29:37 +00:00

test_packed_sequence.py

[4/N] Apply py39 ruff and pyupgrade fixes (#143257)

2025-01-04 10:47:51 +00:00

test_parametrization.py

fix: raise value error on init ParametrizationList if original.device != new.device (#162717)

2025-09-11 23:07:58 +00:00

test_pooling.py

[BUG] MaxUnpool2d/3d should check output dim before accessing its elements (#163507)

2025-09-22 21:36:48 +00:00

test_pruning.py

Fix unused Python variables in test/nn (#143396)

2024-12-18 03:30:54 +00:00