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

Add Half support for aminmax on CPU (#106853)

Browse Source 
Add Half support for aminmax on CPU.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/106853
Approved by: https://github.com/cpuhrsch
This commit is contained in:
CaoE
2023-10-23 17:43:47 +00:00
committed by PyTorch MergeBot
parent ad4ccf9689
commit 4b324a8717
5 changed files with 5 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
test/test_reductions.py
View File

@ -1207,7 +1207,7 @@ class TestReductions(TestCase):
self._test_minmax_helper(torch.amax, np.amax, device, dtype)
@onlyNativeDeviceTypes
@dtypes(torch.float, torch.double)
@dtypes(torch.float, torch.double, torch.bfloat16, torch.half)
@dtypesIfCUDA(torch.half, torch.float, torch.bfloat16)
def test_aminmax(self, device, dtype):