numba-cuda 0.23.0__cp313-cp313-win_amd64.whl → 0.24.0__cp313-cp313-win_amd64.whl

numba_cuda/numba/cuda/core/compiler.py

@@ -23,10 +23,9 @@ class _CompileStatus(object):
         self.can_fallback = can_fallback
 
     def __repr__(self):
-        vals = []
-        for k in self.__slots__:
-            vals.append("{k}={v}".format(k=k, v=getattr(self, k)))
-        return ", ".join(vals)
+        return ", ".join(
+            "{k}={v}".format(k=k, v=getattr(self, k)) for k in self.__slots__
+        )
 
 
 class StateDict(dict):

numba_cuda/numba/cuda/core/compiler_machinery.py

@@ -386,7 +386,7 @@ class PassManager(object):
                     self._runPass(idx, pass_inst, state)
                 else:
                     raise BaseException("Legacy pass in use")
-            except _EarlyPipelineCompletion as e:
+            except _EarlyPipelineCompletion as e:  # noqa: PERF203
                 raise e
             except Exception as e:
                 if not isinstance(e, errors.NumbaError):
@@ -468,14 +468,14 @@ class PassRegistry(object):
         return self._registry[clazz]
 
     def _does_pass_name_alias(self, check):
-        for k, v in self._registry.items():
+        for v in self._registry.values():
             if v.pass_inst.name == check:
                 return True
         return False
 
     def find_by_name(self, class_name):
         assert isinstance(class_name, str)
-        for k, v in self._registry.items():
+        for v in self._registry.values():
             if v.pass_inst.name == class_name:
                 return v
         else:

numba_cuda/numba/cuda/core/config.py

@@ -153,9 +153,11 @@ class _EnvReloader(object):
                         new_environ["NUMBA_" + k.upper()] = v
 
         # clobber file based config with any locally defined env vars
-        for name, value in os.environ.items():
-            if name.startswith("NUMBA_"):
-                new_environ[name] = value
+        new_environ.update(
+            (name, value)
+            for name, value in os.environ.items()
+            if name.startswith("NUMBA_")
+        )
         # We update the config variables if at least one NUMBA environment
         # variable was modified.  This lets the user modify values
         # directly in the config module without having them when
@@ -494,10 +496,6 @@ class _EnvReloader(object):
             "NUMBA_CUDA_PER_THREAD_DEFAULT_STREAM", int, 0
         )
 
-        CUDA_ENABLE_MINOR_VERSION_COMPATIBILITY = _readenv(
-            "NUMBA_CUDA_ENABLE_MINOR_VERSION_COMPATIBILITY", int, 0
-        )
-
         # Location of the CUDA include files
         if IS_WIN32:
             cuda_path = os.environ.get("CUDA_PATH")

numba_cuda/numba/cuda/core/controlflow.py

@@ -11,9 +11,16 @@ from numba.cuda.utils import PYVERSION
 
 # List of bytecodes creating a new block in the control flow graph
 # (in addition to explicit jump labels).
-NEW_BLOCKERS = frozenset(
-    ["SETUP_LOOP", "FOR_ITER", "SETUP_WITH", "BEFORE_WITH"]
-)
+if PYVERSION in ((3, 14),):
+    NEW_BLOCKERS = frozenset(
+        ["SETUP_LOOP", "FOR_ITER", "SETUP_WITH", "BEFORE_WITH", "LOAD_SPECIAL"]
+    )
+elif PYVERSION in ((3, 10), (3, 11), (3, 12), (3, 13)):
+    NEW_BLOCKERS = frozenset(
+        ["SETUP_LOOP", "FOR_ITER", "SETUP_WITH", "BEFORE_WITH"]
+    )
+else:
+    raise NotImplementedError(PYVERSION)
 
 
 class CFBlock(object):
@@ -400,8 +407,7 @@ class CFGraph(object):
             if node not in seen:
                 yield node
                 seen.add(node)
-                for succ in self._succs[node]:
-                    stack.append(succ)
+                stack.extend(self._succs[node])
 
     def _eliminate_dead_blocks(self):
         """
@@ -440,9 +446,11 @@ class CFGraph(object):
             if node not in seen:
                 seen.add(node)
                 stack.append((post_order.append, node))
-                for dest in succs[node]:
-                    if (node, dest) not in back_edges:
-                        stack.append((dfs_rec, dest))
+                stack.extend(
+                    (dfs_rec, dest)
+                    for dest in succs[node]
+                    if (node, dest) not in back_edges
+                )
 
         stack = [(dfs_rec, self._entry_point)]
         while stack:
@@ -970,7 +978,7 @@ class ControlFlowAnalysis(object):
         self._curblock.terminating = True
         self._force_new_block = True
 
-    if PYVERSION in ((3, 12), (3, 13)):
+    if PYVERSION in ((3, 12), (3, 13), (3, 14)):
 
         def op_RETURN_CONST(self, inst):
             self._curblock.terminating = True

numba_cuda/numba/cuda/core/inline_closurecall.py

@@ -344,10 +344,10 @@ class InlineWorker(object):
         # Always copy the callee IR, it gets mutated
         def copy_ir(the_ir):
             kernel_copy = the_ir.copy()
-            kernel_copy.blocks = {}
-            for block_label, block in the_ir.blocks.items():
-                new_block = copy.deepcopy(the_ir.blocks[block_label])
-                kernel_copy.blocks[block_label] = new_block
+            kernel_copy.blocks = {
+                block_label: copy.deepcopy(block)
+                for block_label, block in the_ir.blocks.items()
+            }
             return kernel_copy
 
         callee_ir = copy_ir(callee_ir)
@@ -834,7 +834,7 @@ def _debug_dump(func_ir):
 def _get_all_scopes(blocks):
     """Get all block-local scopes from an IR."""
     all_scopes = []
-    for label, block in blocks.items():
+    for block in blocks.values():
         if block.scope not in all_scopes:
             all_scopes.append(block.scope)
     return all_scopes
@@ -844,7 +844,7 @@ def _replace_args_with(blocks, args):
     """
     Replace ir.Arg(...) with real arguments from call site
     """
-    for label, block in blocks.items():
+    for block in blocks.values():
         assigns = block.find_insts(ir.assign_types)
         for stmt in assigns:
             if isinstance(stmt.value, ir.arg_types):
@@ -857,7 +857,7 @@ def _replace_freevars(blocks, args):
     """
     Replace ir.FreeVar(...) with real variables from parent function
     """
-    for label, block in blocks.items():
+    for block in blocks.values():
         assigns = block.find_insts(ir.assign_types)
         for stmt in assigns:
             if isinstance(stmt.value, ir.freevar_types):
@@ -873,7 +873,7 @@ def _replace_returns(blocks, target, return_label):
     """
     Return return statement by assigning directly to target, and a jump.
     """
-    for label, block in blocks.items():
+    for block in blocks.values():
         casts = []
         for i in range(len(block.body)):
             stmt = block.body[i]
@@ -1316,8 +1316,9 @@ def _inline_arraycall(
     )
 
     # Add back removed just in case they are used by something else
-    for var in removed:
-        stmts.append(_new_definition(func_ir, var, array_var, loc))
+    stmts.extend(
+        _new_definition(func_ir, var, array_var, loc) for var in removed
+    )
 
     # Add back terminator
     stmts.append(terminator)