commit 4b324a8717e54faf1664248c3347633f4ff1e428
author CaoE <e.cao@intel.com> Mon Oct 23 17:43:47 2023 +0000
committer PyTorch MergeBot <pytorchmergebot@users.noreply.github.com> Mon Oct 23 17:43:47 2023 +0000
tree ac9b2c51e7b0b00ccb338b43c331793209123377
parent ad4ccf96896bdf0f098bd9192f8c5a019fddf4c6

Add Half support for aminmax on CPU (#106853)

Add Half support for aminmax on CPU.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/106853
Approved by: https://github.com/cpuhrsch

test/test_mps.py

diff --git a/test/test_mps.py b/test/test_mps.py
index e6da181..5d3a19e 100644
--- a/test/test_mps.py
+++ b/test/test_mps.py
@@ -79,7 +79,7 @@
         'cdist': [torch.float32],
         'masked.scatter': [torch.float16, torch.float32],
         'index_fill': [torch.float16, torch.float32],  # missing `aten::_unique`.
-        'aminmax': [torch.float32],
+        'aminmax': [torch.float32, torch.float16],
         'polar': [torch.float32],
 
         # Correctness issues