Skip to content

Commit f92751e

Browse files
committed
nit
1 parent 0afebbd commit f92751e

File tree

1 file changed

+6
-1
lines changed

1 file changed

+6
-1
lines changed

numba_cuda/numba/cuda/fp16.py

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -185,11 +185,16 @@ def log10_ol(a):
185185
return _make_unary(a, hlog10)
186186

187187

188-
@overload(, target="cuda")
188+
@overload(math.exp, target="cuda")
189189
def exp_ol(a):
190190
return _make_unary(a, hexp)
191191

192192

193+
@overload(math.exp2, target="cuda")
194+
def exp2_ol(a):
195+
return _make_unary(a, hexp2)
196+
197+
193198
@overload(math.tanh, target="cuda")
194199
def tanh_ol(a):
195200
return _make_unary(a, htanh)

0 commit comments

Comments
 (0)