Merge pull request #9416 from guilhermeleobas/guilhermeleobas/cpu_mat…

…h_log2

Add math.log2 support

docs/source/reference/pysupported.rst

@@ -1074,6 +1074,7 @@ The following functions from the :mod:`math` module are supported:
 * :func:`math.ldexp`
 * :func:`math.lgamma`
 * :func:`math.log`
+* :func:`math.log2`
 * :func:`math.log10`
 * :func:`math.log1p`
 * :func:`math.pow`

docs/upcoming_changes/9416.new_feature.rst

@@ -0,0 +1,4 @@
+Add support for ``math.log2``.
+------------------------------
+
+Support for ``math.log2`` is added.

numba/core/typing/mathdecl.py

@@ -14,6 +14,7 @@
 @infer_global(math.log)
 @infer_global(math.log1p)
 @infer_global(math.log10)
+@infer_global(math.log2)
 @infer_global(math.sin)
 @infer_global(math.cos)
 @infer_global(math.tan)

numba/cpython/mathimpl.py

@@ -176,6 +176,7 @@ def float_impl(context, builder, sig, args):
 exp_impl = unary_math_intr(math.exp, 'llvm.exp')
 log_impl = unary_math_intr(math.log, 'llvm.log')
 log10_impl = unary_math_intr(math.log10, 'llvm.log10')
+log2_impl = unary_math_intr(math.log2, 'llvm.log2')
 sin_impl = unary_math_intr(math.sin, 'llvm.sin')
 cos_impl = unary_math_intr(math.cos, 'llvm.cos')
 

numba/tests/test_mathlib.py

@@ -91,6 +91,10 @@ def log10(x):
     return math.log10(x)
 
 
+def log2(x):
+    return math.log2(x)
+
+
 def floor(x):
     return math.floor(x)
 
@@ -285,6 +289,14 @@ def test_log10(self):
         x_values = [1, 10, 100, 1000, 100000, 1000000, 0.1, 1.1]
         self.run_unary(pyfunc, x_types, x_values)
 
+    def test_log2(self):
+        pyfunc = log2
+        x_types = [types.int16, types.int32, types.int64,
+                   types.uint16, types.uint32, types.uint64,
+                   types.float32, types.float64]
+        x_values = [1, 10, 100, 1000, 100000, 1000000, 0.1, 1.1]
+        self.run_unary(pyfunc, x_types, x_values)
+
     def test_asin(self):
         pyfunc = asin
         x_types = [types.int16, types.int32, types.int64,