Add implementations for math.trunc for float32/float64 for cuda target

Deezy33-creator · Nov 21, 2022 · 25de1a3 · 25de1a3
1 parent 2f01c9a
commit 25de1a3
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/numba/cuda/mathimpl.py b/numba/cuda/mathimpl.py
@@ -43,6 +43,7 @@
 unarys += [('atanh', 'atanhf', math.atanh)]
 unarys += [('tan', 'tanf', math.tan)]
 unarys += [('tanh', 'tanhf', math.tanh)]
+unarys += [('trunc', 'truncf', math.trunc)]
 
 unarys_fastmath = {}
 unarys_fastmath['cosf'] = 'fast_cosf'

diff --git a/numba/cuda/tests/cudapy/test_math.py b/numba/cuda/tests/cudapy/test_math.py
@@ -652,6 +652,15 @@ def test_math_floor(self):
         self.unary_template_int64(math_floor, np.floor)
         self.unary_template_uint64(math_floor, np.floor)
 
+    #---------------------------------------------------------------------------
+    # test_math_trunc
+
+    def test_math_trunc(self):
+        self.unary_template_float32(math_trunc, np.trunc)
+        self.unary_template_float64(math_trunc, np.trunc)
+        self.unary_template_int64(math_trunc, np.trunc)
+        self.unary_template_uint64(math_trunc, np.trunc)
+
     #---------------------------------------------------------------------------
     # test_math_copysign