Merge pull request #1435 from IntelPython/fix/windows-parfor

Migrate parfor to SPIRVKernelDispatcher 675cb79

Author: github-actions[bot]

dev/.buildinfo

@@ -1,4 +1,4 @@
 # Sphinx build info version 1
 # This file hashes the configuration used when building these files. When it is not found, a full rebuild will be done.
-config: 4dd9bd111271e8299ebcca45d149268b
+config: fd05f27b09982a018355b52b5e35eb29
 tags: 645f666f9bcd5a90fca523b33c5a78b7

dev/_sources/autoapi/numba_dpex/core/parfors/kernel_builder/index.rst.txt

@@ -26,7 +26,7 @@ Overview
    :widths: auto
    :class: summarytable
 
-   * - :py:obj:`create_kernel_for_parfor <numba_dpex.core.parfors.kernel_builder.create_kernel_for_parfor>`\ (lowerer, parfor_node, typemap, flags, loop_ranges, has_aliases, races, parfor_outputs)
+   * - :py:obj:`create_kernel_for_parfor <numba_dpex.core.parfors.kernel_builder.create_kernel_for_parfor>`\ (lowerer, parfor_node, typemap, loop_ranges, races, parfor_outputs)
      - Creates a numba_dpex.kernel function for a parfor node.
    * - :py:obj:`update_sentinel <numba_dpex.core.parfors.kernel_builder.update_sentinel>`\ (kernel_ir, sentinel_name, kernel_body, new_label)
      - Searched all the blocks in the IR generated from a kernel template and
@@ -36,14 +36,14 @@ Overview
 Classes
 -------
 
-.. py:class:: ParforKernel(name, kernel, signature, kernel_args, kernel_arg_types, queue: dpctl.SyclQueue, local_accessors=None, work_group_size=None)
+.. py:class:: ParforKernel(signature, kernel_args, kernel_arg_types, local_accessors=None, work_group_size=None, kernel_module=None)
 
 
 
 
 Functions
 ---------
-.. py:function:: create_kernel_for_parfor(lowerer, parfor_node, typemap, flags, loop_ranges, has_aliases, races, parfor_outputs) -> ParforKernel
+.. py:function:: create_kernel_for_parfor(lowerer, parfor_node, typemap, loop_ranges, races, parfor_outputs) -> ParforKernel
 
    Creates a numba_dpex.kernel function for a parfor node.
 

dev/_sources/autoapi/numba_dpex/core/parfors/kernel_templates/kernel_template_iface/index.rst.txt

@@ -39,8 +39,6 @@ Classes
 
       * - :py:obj:`dump_kernel_string <numba_dpex.core.parfors.kernel_templates.kernel_template_iface.KernelTemplateInterface.dump_kernel_string>`\ ()
         - :summarylabel:`abc` \-
-      * - :py:obj:`dump_kernel_ir <numba_dpex.core.parfors.kernel_templates.kernel_template_iface.KernelTemplateInterface.dump_kernel_ir>`\ ()
-        - :summarylabel:`abc` \-
 
 
    .. rubric:: Members
@@ -49,10 +47,6 @@ Classes
       :abstractmethod:
 
 
-   .. py:method:: dump_kernel_ir()
-      :abstractmethod:
-
-
 
 
 

dev/_sources/autoapi/numba_dpex/core/parfors/kernel_templates/range_kernel_template/index.rst.txt

@@ -42,8 +42,6 @@ Classes
 
       * - :py:obj:`dump_kernel_string <numba_dpex.core.parfors.kernel_templates.range_kernel_template.RangeKernelTemplate.dump_kernel_string>`\ ()
         - Helper to print the kernel function string.
-      * - :py:obj:`dump_kernel_ir <numba_dpex.core.parfors.kernel_templates.range_kernel_template.RangeKernelTemplate.dump_kernel_ir>`\ ()
-        - Helper to dump the Numba IR for the RangeKernelTemplate.
 
 
    .. rubric:: Members
@@ -53,11 +51,6 @@ Classes
       Helper to print the kernel function string.
 
 
-   .. py:method:: dump_kernel_ir()
-
-      Helper to dump the Numba IR for the RangeKernelTemplate.
-
-
 
 
 

dev/_sources/autoapi/numba_dpex/core/parfors/kernel_templates/reduction_template/index.rst.txt

@@ -46,8 +46,6 @@ Classes
 
       * - :py:obj:`dump_kernel_string <numba_dpex.core.parfors.kernel_templates.reduction_template.TreeReduceIntermediateKernelTemplate.dump_kernel_string>`\ ()
         - Helper to print the kernel function string.
-      * - :py:obj:`dump_kernel_ir <numba_dpex.core.parfors.kernel_templates.reduction_template.TreeReduceIntermediateKernelTemplate.dump_kernel_ir>`\ ()
-        - Helper to dump the Numba IR for a
 
 
    .. rubric:: Members
@@ -57,12 +55,6 @@ Classes
       Helper to print the kernel function string.
 
 
-   .. py:method:: dump_kernel_ir()
-
-      Helper to dump the Numba IR for a
-      TreeReduceIntermediateKernelTemplate.
-
-
 
 
 .. py:class:: RemainderReduceIntermediateKernelTemplate(kernel_name, kernel_params, sentinel_name, redvars, parfor_reddict, redvars_dict, typemap, legal_loop_indices, global_size_var_name, global_size_mod_var_name, partial_sum_size_var_name, partial_sum_var_name, final_sum_var_name, reductionKernelVar)
@@ -83,8 +75,6 @@ Classes
 
       * - :py:obj:`dump_kernel_string <numba_dpex.core.parfors.kernel_templates.reduction_template.RemainderReduceIntermediateKernelTemplate.dump_kernel_string>`\ ()
         - Helper to print the kernel function string.
-      * - :py:obj:`dump_kernel_ir <numba_dpex.core.parfors.kernel_templates.reduction_template.RemainderReduceIntermediateKernelTemplate.dump_kernel_ir>`\ ()
-        - Helper to dump the Numba IR for the
 
 
    .. rubric:: Members
@@ -94,12 +84,6 @@ Classes
       Helper to print the kernel function string.
 
 
-   .. py:method:: dump_kernel_ir()
-
-      Helper to dump the Numba IR for the
-      RemainderReduceIntermediateKernelTemplate.
-
-
 
 
 

dev/_sources/autoapi/numba_dpex/core/parfors/parfor_lowerer/index.rst.txt

@@ -24,14 +24,6 @@ Overview
 
 
 
-.. list-table:: Attributes
-   :header-rows: 0
-   :widths: auto
-   :class: summarytable
-
-   * - :py:obj:`keep_alive_kernels <numba_dpex.core.parfors.parfor_lowerer.keep_alive_kernels>`
-     - \-
-
 
 Classes
 -------
@@ -96,12 +88,5 @@ Classes
 
 
 
-Attributes
-----------
-.. py:data:: keep_alive_kernels
-   :value: []
-
-   
-
 
 

dev/_sources/autoapi/numba_dpex/core/parfors/parfor_pass/index.rst.txt

@@ -36,6 +36,8 @@ Overview
 
    * - :py:obj:`get_dpnp_ufunc_typ <numba_dpex.core.parfors.parfor_pass.get_dpnp_ufunc_typ>`\ (func)
      - get type of the incoming function from builtin registry
+   * - :py:obj:`mk_alloc <numba_dpex.core.parfors.parfor_pass.mk_alloc>`\ (typingctx, typemap, calltypes, lhs, size_var, dtype, scope, loc, lhs_typ, \*\*kws)
+     - generate an array allocation with np.empty() and return list of nodes.
 
 
 
@@ -93,6 +95,15 @@ Functions
    get type of the incoming function from builtin registry
 
 
+.. py:function:: mk_alloc(typingctx, typemap, calltypes, lhs, size_var, dtype, scope, loc, lhs_typ, **kws)
+
+   generate an array allocation with np.empty() and return list of nodes.
+   size_var can be an int variable or tuple of int variables.
+   lhs_typ is the type of the array being allocated.
+
+   Taken from numba, added kws argument to pass it to __allocate__
+
+
 
 
 

dev/_sources/autoapi/numba_dpex/core/parfors/parfor_sentinel_replace_pass/index.rst.txt

@@ -0,0 +1,105 @@
+
+
+:orphan:
+
+numba_dpex.core.parfors.parfor_sentinel_replace_pass
+====================================================
+
+.. py:module:: numba_dpex.core.parfors.parfor_sentinel_replace_pass
+
+
+
+
+Overview
+--------
+.. list-table:: Classes
+   :header-rows: 0
+   :widths: auto
+   :class: summarytable
+
+   * - :py:obj:`ParforBodyArguments <numba_dpex.core.parfors.parfor_sentinel_replace_pass.ParforBodyArguments>`
+     - Arguments containing information to inject parfor code inside kernel.
+   * - :py:obj:`ParforSentinelReplacePass <numba_dpex.core.parfors.parfor_sentinel_replace_pass.ParforSentinelReplacePass>`
+     - Base class for function passes
+
+
+
+
+Classes
+-------
+
+.. py:class:: ParforBodyArguments
+
+   Bases: :py:obj:`NamedTuple`
+
+   Arguments containing information to inject parfor code inside kernel.
+
+
+   .. rubric:: Overview
+
+   .. list-table:: Attributes
+      :header-rows: 0
+      :widths: auto
+      :class: summarytable
+
+      * - :py:obj:`loop_body <numba_dpex.core.parfors.parfor_sentinel_replace_pass.ParforBodyArguments.loop_body>`
+        - \-
+      * - :py:obj:`param_dict <numba_dpex.core.parfors.parfor_sentinel_replace_pass.ParforBodyArguments.param_dict>`
+        - \-
+      * - :py:obj:`legal_loop_indices <numba_dpex.core.parfors.parfor_sentinel_replace_pass.ParforBodyArguments.legal_loop_indices>`
+        - \-
+
+
+
+   .. rubric:: Members
+
+   .. py:attribute:: loop_body
+      :type: any
+
+      
+
+   .. py:attribute:: param_dict
+      :type: any
+
+      
+
+   .. py:attribute:: legal_loop_indices
+      :type: any
+
+      
+
+
+
+.. py:class:: ParforSentinelReplacePass
+
+   Bases: :py:obj:`numba.core.compiler_machinery.FunctionPass`
+
+   Base class for function passes
+
+
+
+   .. rubric:: Overview
+
+
+   .. list-table:: Methods
+      :header-rows: 0
+      :widths: auto
+      :class: summarytable
+
+      * - :py:obj:`run_pass <numba_dpex.core.parfors.parfor_sentinel_replace_pass.ParforSentinelReplacePass.run_pass>`\ (state)
+        - Runs the pass itself. Must return True/False depending on whether
+
+
+   .. rubric:: Members
+
+   .. py:method:: run_pass(state)
+
+      Runs the pass itself. Must return True/False depending on whether
+      statement level modification took place.
+
+
+
+
+
+
+

dev/_sources/autoapi/numba_dpex/core/parfors/reduction_helper/index.rst.txt

@@ -51,13 +51,13 @@ Classes
       :widths: auto
       :class: summarytable
 
-      * - :py:obj:`copy_final_sum_to_host <numba_dpex.core.parfors.reduction_helper.ReductionKernelVariables.copy_final_sum_to_host>`\ (parfor_kernel)
+      * - :py:obj:`copy_final_sum_to_host <numba_dpex.core.parfors.reduction_helper.ReductionKernelVariables.copy_final_sum_to_host>`\ (queue_ref)
         - \-
 
 
    .. rubric:: Members
 
-   .. py:method:: copy_final_sum_to_host(parfor_kernel)
+   .. py:method:: copy_final_sum_to_host(queue_ref)
 
 
 

dev/_sources/autoapi/numba_dpex/core/parfors/reduction_kernel_builder/index.rst.txt

@@ -18,22 +18,22 @@ Overview
    :widths: auto
    :class: summarytable
 
-   * - :py:obj:`create_reduction_main_kernel_for_parfor <numba_dpex.core.parfors.reduction_kernel_builder.create_reduction_main_kernel_for_parfor>`\ (loop_ranges, parfor_node, typemap, flags, has_aliases, reductionKernelVar, parfor_reddict)
+   * - :py:obj:`create_reduction_main_kernel_for_parfor <numba_dpex.core.parfors.reduction_kernel_builder.create_reduction_main_kernel_for_parfor>`\ (loop_ranges, parfor_node, typemap, reductionKernelVar, parfor_reddict)
      - Creates a numba_dpex.kernel function for reduction main kernel.
-   * - :py:obj:`create_reduction_remainder_kernel_for_parfor <numba_dpex.core.parfors.reduction_kernel_builder.create_reduction_remainder_kernel_for_parfor>`\ (parfor_node, typemap, flags, has_aliases, reductionKernelVar, parfor_reddict, reductionHelperList)
+   * - :py:obj:`create_reduction_remainder_kernel_for_parfor <numba_dpex.core.parfors.reduction_kernel_builder.create_reduction_remainder_kernel_for_parfor>`\ (parfor_node, typemap, reductionKernelVar, parfor_reddict, reductionHelperList)
      - Creates a numba_dpex.kernel function for a reduction remainder kernel.
 
 
 
 
 Functions
 ---------
-.. py:function:: create_reduction_main_kernel_for_parfor(loop_ranges, parfor_node, typemap, flags, has_aliases, reductionKernelVar: numba_dpex.core.parfors.reduction_helper.ReductionKernelVariables, parfor_reddict=None)
+.. py:function:: create_reduction_main_kernel_for_parfor(loop_ranges, parfor_node, typemap, reductionKernelVar: numba_dpex.core.parfors.reduction_helper.ReductionKernelVariables, parfor_reddict=None)
 
    Creates a numba_dpex.kernel function for reduction main kernel.
 
 
-.. py:function:: create_reduction_remainder_kernel_for_parfor(parfor_node, typemap, flags, has_aliases, reductionKernelVar, parfor_reddict, reductionHelperList)
+.. py:function:: create_reduction_remainder_kernel_for_parfor(parfor_node, typemap, reductionKernelVar, parfor_reddict, reductionHelperList)
 
    Creates a numba_dpex.kernel function for a reduction remainder kernel.