PyTorch MergeBot fa8e3a28a7 Revert "[cuDNN][SDPA][Nested Tensor] Experimental cuDNN Nested Tensor SDPA Support (forward only) (#141178)" ... This reverts commit 533b884870. Reverted https://github.com/pytorch/pytorch/pull/141178 on behalf of https://github.com/jeanschmidt due to Broke internal arvr signals, see D69971019. @jbschlosser please help the author get this PR merged ([comment](https://github.com/pytorch/pytorch/pull/141178#issuecomment-2676317470))		2025-02-22 17:28:12 +00:00
..
__init__.py
nested_int.py	Switch to using Python nested int (#141166)	2024-12-02 19:17:30 +00:00
nested_tensor.py	[NJT] Fix inference mode for composite implicit ops without nested-specific kernel (#146633)	2025-02-10 16:59:48 +00:00
ops.py	Small improvements to NJT matrix multiplies (#146405)	2025-02-06 04:51:12 +00:00
sdpa.py	Revert "[cuDNN][SDPA][Nested Tensor] Experimental cuDNN Nested Tensor SDPA Support (forward only) (#141178)"	2025-02-22 17:28:12 +00:00