Description
🐛 Describe the bug
TestReductionsCUDA.test_reductions_large_half_tensors_cuda_complex32 fails with "mean_cuda" not implemented for 'ComplexHalf'
root@9dd3e47db792:/opt/pytorch/pytorch# python test/test_reductions.py -v -k TestReductionsCUDA.test_reductions_large_half_tensors_cuda_complex32
test_reductions_large_half_tensors_cuda_complex32 (__main__.TestReductionsCUDA) ... test/test_reductions.py:3390: UserWarning: ComplexHalf support is experimental and many operators don't support it yet. (Triggered internally at /opt/pytorch/pytorch/aten/src/ATen/EmptyTensor.cpp:31.)
t = torch.ones(2**31, device=device, dtype=dtype)
ERROR
======================================================================
ERROR: test_reductions_large_half_tensors_cuda_complex32 (__main__.TestReductionsCUDA)
----------------------------------------------------------------------
Traceback (most recent call last):
File "/opt/pytorch/pytorch/torch/testing/_internal/common_utils.py", line 1990, in wrapper
method(*args, **kwargs)
File "/opt/pytorch/pytorch/torch/testing/_internal/common_utils.py", line 1990, in wrapper
method(*args, **kwargs)
File "/opt/pytorch/pytorch/torch/testing/_internal/common_device_type.py", line 391, in instantiated_test
raise rte
File "/opt/pytorch/pytorch/torch/testing/_internal/common_device_type.py", line 378, in instantiated_test
result = test(self, **param_kwargs)
File "/opt/pytorch/pytorch/torch/testing/_internal/common_device_type.py", line 984, in only_fn
return fn(slf, *args, **kwargs)
File "/opt/pytorch/pytorch/torch/testing/_internal/common_device_type.py", line 945, in dep_fn
return fn(self, *args, **kwargs)
File "test/test_reductions.py", line 3402, in test_reductions_large_half_tensors
self.assertEqual(torch.mean(t), expected)
RuntimeError: "mean_cuda" not implemented for 'ComplexHalf'
----------------------------------------------------------------------
Ran 1 test in 2.856s
FAILED (errors=1)Versions
failed: 1707438
passed: c6b7c33