Skip to content

Commit 22be5ee

Browse files
committed
fix: test
1 parent 1a8564a commit 22be5ee

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

numba_cuda/numba/cuda/tests/cudapy/test_fastmath.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,7 @@
77
from numba import cuda
88
from numba.cuda import float32
99
from numba.cuda.compiler import compile_ptx_for_current_device, compile_ptx
10-
from math import cos, sin, tan, exp, exp2, log, log10, log2, pow, tanh
10+
from math import cos, sin, tan, exp, log, log10, log2, pow, tanh
1111
from operator import truediv
1212
import numpy as np
1313
from numba.cuda.testing import CUDATestCase, skip_on_cudasim, skip_unless_cc_75
@@ -145,6 +145,7 @@ def test_expf(self):
145145

146146
@unittest.skipUnless(sys.version_info >= (3, 11), "Python 3.11+ required")
147147
def test_exp2f(self):
148+
from math import exp2
148149
self._test_fast_math_unary(
149150
exp2,
150151
FastMathCriterion(

0 commit comments

Comments
 (0)