Jeff Daily e3aefe2970 Revert "Initial Flash Attention support on ROCM (#114309)" (#115975) ... This reverts commit 5bddbed399. Pull Request resolved: https://github.com/pytorch/pytorch/pull/115975 Approved by: https://github.com/atalman, https://github.com/malfet		2023-12-16 03:40:14 +00:00
..
_internal	Revert "Initial Flash Attention support on ROCM (#114309)" (#115975)	2023-12-16 03:40:14 +00:00
__init__.py	document torch.testing.assert_allclose (#89526)	2022-12-01 11:22:50 +00:00
_comparison.py	[BE]: Update lintrunner mypy to 1.6.0 (#111375)	2023-10-17 01:22:06 +00:00
_creation.py	Add meta func for scaled mm (#112609)	2023-11-03 03:44:22 +00:00