numba-cuda 0.8.0__py3-none-any.whl → 0.9.0__py3-none-any.whl

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

@@ -8,88 +8,82 @@ def template(signature, shapes, expects):
     for k, v in expects.items():
         got = getattr(sch, k)
         if got != v:
-            fmt = 'error for %s: got=%s but expect=%s'
+            fmt = "error for %s: got=%s but expect=%s"
             raise AssertionError(fmt % (k, got, v))
 
 
 class TestGUFuncScheduling(unittest.TestCase):
     def test_signature_1(self):
-        signature = '(m, n), (n, p) -> (m, p)'
+        signature = "(m, n), (n, p) -> (m, p)"
         shapes = (100, 4, 5), (1, 5, 7)
         expects = dict(
             ishapes=[(4, 5), (5, 7)],
             oshapes=[(4, 7)],
             loopdims=(100,),
-            pinned=[False, True]
+            pinned=[False, True],
         )
         template(signature, shapes, expects)
 
     def test_signature_2(self):
-        signature = '(m, n), (n, p) -> (m, p)'
+        signature = "(m, n), (n, p) -> (m, p)"
         shapes = (100, 4, 5), (100, 5, 7)
         expects = dict(
             ishapes=[(4, 5), (5, 7)],
             oshapes=[(4, 7)],
             loopdims=(100,),
-            pinned=[False, False]
+            pinned=[False, False],
         )
         template(signature, shapes, expects)
 
     def test_signature_3(self):
-        signature = '(m, n), (n, p) -> (m, p)'
+        signature = "(m, n), (n, p) -> (m, p)"
         shapes = (12, 34, 4, 5), (12, 34, 5, 7)
         expects = dict(
             ishapes=[(4, 5), (5, 7)],
             oshapes=[(4, 7)],
             loopdims=(12, 34),
-            pinned=[False, False]
+            pinned=[False, False],
         )
         template(signature, shapes, expects)
 
     def test_signature_4(self):
-        signature = '(m, n), (n, p) -> (m, p)'
+        signature = "(m, n), (n, p) -> (m, p)"
         shapes = (4, 5), (5, 7)
         expects = dict(
             ishapes=[(4, 5), (5, 7)],
             oshapes=[(4, 7)],
             loopdims=(),
-            pinned=[False, False]
+            pinned=[False, False],
         )
         template(signature, shapes, expects)
 
     def test_signature_5(self):
-        signature = '(a), (a) -> (a)'
+        signature = "(a), (a) -> (a)"
         shapes = (5,), (5,)
         expects = dict(
             ishapes=[(5,), (5,)],
             oshapes=[(5,)],
             loopdims=(),
-            pinned=[False, False]
+            pinned=[False, False],
         )
         template(signature, shapes, expects)
 
     def test_signature_6(self):
-        signature = '(), () -> ()'
+        signature = "(), () -> ()"
         shapes = (5,), (5,)
         expects = dict(
-            ishapes=[(), ()],
-            oshapes=[()],
-            loopdims=(5,),
-            pinned=[False, False]
+            ishapes=[(), ()], oshapes=[()], loopdims=(5,), pinned=[False, False]
         )
         template(signature, shapes, expects)
 
     def test_signature_7(self):
-        signature = '(), () -> ()'
+        signature = "(), () -> ()"
         shapes = (5,), ()
         expects = dict(
-            ishapes=[(), ()],
-            oshapes=[()],
-            loopdims=(5,),
-            pinned=[False, True]
+            ishapes=[(), ()], oshapes=[()], loopdims=(5,), pinned=[False, True]
         )
         template(signature, shapes, expects)
 
 
-if __name__ == '__main__':
+if __name__ == "__main__":
     unittest.main()

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

@@ -5,7 +5,6 @@ from numba.cuda.testing import unittest, CUDATestCase
 
 class TestCudaIDiv(CUDATestCase):
     def test_inplace_div(self):
-
         @cuda.jit(void(float32[:, :], int32, int32))
         def div(grid, l_x, l_y):
             for x in range(l_x):
@@ -19,7 +18,6 @@ class TestCudaIDiv(CUDATestCase):
         self.assertTrue(np.all(y == 0.5))
 
     def test_inplace_div_double(self):
-
         @cuda.jit(void(float64[:, :], int32, int32))
         def div_double(grid, l_x, l_y):
             for x in range(l_x):
@@ -33,5 +31,5 @@ class TestCudaIDiv(CUDATestCase):
         self.assertTrue(np.all(y == 0.5))
 
 
-if __name__ == '__main__':
+if __name__ == "__main__":
     unittest.main()

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

@@ -3,11 +3,14 @@ import numpy as np
 from io import StringIO
 from numba import cuda, float32, float64, int32, intp
 from numba.cuda.testing import unittest, CUDATestCase
-from numba.cuda.testing import (skip_on_cudasim, skip_with_nvdisasm,
-                                skip_without_nvdisasm)
+from numba.cuda.testing import (
+    skip_on_cudasim,
+    skip_with_nvdisasm,
+    skip_without_nvdisasm,
+)
 
 
-@skip_on_cudasim('Simulator does not generate code to be inspected')
+@skip_on_cudasim("Simulator does not generate code to be inspected")
 class TestInspect(CUDATestCase):
     @property
     def cc(self):
@@ -60,7 +63,10 @@ class TestInspect(CUDATestCase):
 
         # Signature in LLVM dict
         llvmirs = foo.inspect_llvm()
-        self.assertEqual(2, len(llvmirs), )
+        self.assertEqual(
+            2,
+            len(llvmirs),
+        )
         self.assertIn((intp, intp), llvmirs)
         self.assertIn((float64, float64), llvmirs)
 
@@ -75,7 +81,10 @@ class TestInspect(CUDATestCase):
         asmdict = foo.inspect_asm()
 
         # Signature in assembly dict
-        self.assertEqual(2, len(asmdict), )
+        self.assertEqual(
+            2,
+            len(asmdict),
+        )
         self.assertIn((intp, intp), asmdict)
         self.assertIn((float64, float64), asmdict)
 
@@ -87,7 +96,7 @@ class TestInspect(CUDATestCase):
         # Ensure function appears in output
         seen_function = False
         for line in sass.split():
-            if '.text' in line and name in line:
+            if ".text" in line and name in line:
                 seen_function = True
         self.assertTrue(seen_function)
 
@@ -95,11 +104,11 @@ class TestInspect(CUDATestCase):
 
         # Some instructions common to all supported architectures that should
         # appear in the output
-        self.assertIn('S2R', sass)   # Special register to register
-        self.assertIn('BRA', sass)   # Branch
-        self.assertIn('EXIT', sass)  # Exit program
+        self.assertIn("S2R", sass)  # Special register to register
+        self.assertIn("BRA", sass)  # Branch
+        self.assertIn("EXIT", sass)  # Exit program
 
-    @skip_without_nvdisasm('nvdisasm needed for inspect_sass()')
+    @skip_without_nvdisasm("nvdisasm needed for inspect_sass()")
     def test_inspect_sass_eager(self):
         sig = (float32[::1], int32[::1])
 
@@ -109,9 +118,9 @@ class TestInspect(CUDATestCase):
             if i < len(x):
                 x[i] += y[i]
 
-        self._test_inspect_sass(add, 'add', add.inspect_sass(sig))
+        self._test_inspect_sass(add, "add", add.inspect_sass(sig))
 
-    @skip_without_nvdisasm('nvdisasm needed for inspect_sass()')
+    @skip_without_nvdisasm("nvdisasm needed for inspect_sass()")
     def test_inspect_sass_lazy(self):
         @cuda.jit(lineinfo=True)
         def add(x, y):
@@ -124,10 +133,11 @@ class TestInspect(CUDATestCase):
         add[1, 10](x, y)
 
         signature = (int32[::1], float32[::1])
-        self._test_inspect_sass(add, 'add', add.inspect_sass(signature))
+        self._test_inspect_sass(add, "add", add.inspect_sass(signature))
 
-    @skip_with_nvdisasm('Missing nvdisasm exception only generated when it is '
-                        'not present')
+    @skip_with_nvdisasm(
+        "Missing nvdisasm exception only generated when it is not present"
+    )
     def test_inspect_sass_nvdisasm_missing(self):
         @cuda.jit((float32[::1],))
         def f(x):
@@ -136,9 +146,9 @@ class TestInspect(CUDATestCase):
         with self.assertRaises(RuntimeError) as raises:
             f.inspect_sass()
 
-        self.assertIn('nvdisasm has not been found', str(raises.exception))
+        self.assertIn("nvdisasm has not been found", str(raises.exception))
 
-    @skip_without_nvdisasm('nvdisasm needed for inspect_sass_cfg()')
+    @skip_without_nvdisasm("nvdisasm needed for inspect_sass_cfg()")
     def test_inspect_sass_cfg(self):
         sig = (float32[::1], int32[::1])
 
@@ -149,10 +159,9 @@ class TestInspect(CUDATestCase):
                 x[i] += y[i]
 
         self.assertRegex(
-            add.inspect_sass_cfg(signature=sig),
-            r'digraph\s*\w\s*{(.|\n)*\n}'
+            add.inspect_sass_cfg(signature=sig), r"digraph\s*\w\s*{(.|\n)*\n}"
         )
 
 
-if __name__ == '__main__':
+if __name__ == "__main__":
     unittest.main()