Add Half support for cummax, cummin, cumprod, logcumsumexp, and prod on CPU (#112132)
Add Half support for cummax, cummin, cumprod, logcumsumexp, and prod on CPU.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/112132
Approved by: https://github.com/cpuhrsch
diff --git a/test/test_mps.py b/test/test_mps.py
index b359a4c..817e11e 100644
--- a/test/test_mps.py
+++ b/test/test_mps.py
@@ -164,7 +164,9 @@
'__rpow__': [torch.float32],
# See https://github.com/pytorch/pytorch/issues/106112 for more information
- 'cumprod': [torch.float32],
+ 'cumprod': [torch.float32, torch.float16],
+ # See https://github.com/pytorch/pytorch/issues/109166 for more information
+ 'masked.cumprod': [torch.float16],
}
SKIPLIST_GRAD = {
@@ -10943,6 +10945,7 @@
'nn.functional.kl_div',
'nn.functional.softmin',
'cross', 'linalg.cross',
+ 'prod', 'masked.prod',
# for macOS 12
'masked.normalize', 'masked.sum', 'masked.var',
