frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

Revert "[MPS] Expand fused forloop to bfloat16 (#141104)"

Browse Source 
This reverts commit 9a729390420570cd2528ce2e9947e3eab209660b.

Reverted https://github.com/pytorch/pytorch/pull/141104 on behalf of https://github.com/malfet due to Want to add test script to the commit message ([comment](https://github.com/pytorch/pytorch/pull/141104#issuecomment-2492659931))
This commit is contained in:
PyTorch MergeBot
2024-11-22 01:03:43 +00:00
parent e8de8f3969
commit 989888236e
3 changed files with 33 additions and 56 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
test/test_optim.py
View File

@ -1027,11 +1027,8 @@ class TestOptimRenewed(TestCase):
if _get_device_type(device) == "mps" and dtype not in (
torch.float16,
torch.float32,
torch.bfloat16,
):
self.skipTest(
"MPS supports only torch.float16, torch.float32 and torch.bfloat16"
)
self.skipTest("MPS supports only torch.float16 and torch.float32")
self._test_derived_optimizers(device, dtype, optim_info, "fused")
@optims(