We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 5cef8fb commit bf55d44Copy full SHA for bf55d44
tests/tensor/utils.py
@@ -152,7 +152,7 @@ def upcast_float16_ufunc(fn):
152
"""
153
154
def ret(*args, **kwargs):
155
- out_dtype = np.find_common_type([a.dtype for a in args], [np.float16])
+ out_dtype = np.result_type(np.float16, *args)
156
if out_dtype == "float16":
157
# Force everything to float32
158
sig = "f" * fn.nin + "->" + "f" * fn.nout
0 commit comments