Merge pull request #1290 from IntelPython/feature/use_dpnp_specific_p…

…arfor

Feature/use dpnp specific parfor b6a13b0

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: 5ce5e8be9715c4f3cb77525aad77bc3a
+config: 77c7ed92725ec04ee213bdd81322e999
 tags: 645f666f9bcd5a90fca523b33c5a78b7

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

@@ -0,0 +1,96 @@
+
+:orphan:
+
+numba_dpex.core.parfors.parfor_pass
+===================================
+
+.. py:module:: numba_dpex.core.parfors.parfor_pass
+
+.. autoapi-nested-parse::
+
+   This module follows the logic of numba/parfors/parfor.py with changes required
+   to use it with dpnp instead of numpy.
+
+
+
+Overview
+--------
+
+.. list-table:: Classes
+   :header-rows: 0
+   :widths: auto
+   :class: summarytable
+
+   * - :py:obj:`ConvertDPNPPass <numba_dpex.core.parfors.parfor_pass.ConvertDPNPPass>`
+     - Convert supported Dpnp functions, as well as arrayexpr nodes, to
+   * - :py:obj:`ParforPass <numba_dpex.core.parfors.parfor_pass.ParforPass>`
+     - Based on the NumpyParforPass. Lot's of code was copy-pasted, with minor
+
+
+.. list-table:: Function
+   :header-rows: 0
+   :widths: auto
+   :class: summarytable
+
+   * - :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
+
+
+
+Classes
+-------
+
+.. py:class:: ConvertDPNPPass(pass_states)
+
+   Bases: :py:obj:`numba.parfors.parfor.ConvertNumpyPass`
+
+   Convert supported Dpnp functions, as well as arrayexpr nodes, to
+   parfor nodes.
+
+   Based on the ConvertNumpyPass. Lot's of code was copy-pasted, with minor
+   changes due to lack of extensibility of the original package.
+
+
+
+
+.. py:class:: ParforPass
+
+   Bases: :py:obj:`numba.core.typed_passes.ParforPass`
+
+   Based on the NumpyParforPass. Lot's of code was copy-pasted, with minor
+   changes due to lack of extensibility of the original package.
+
+
+   .. rubric:: Overview
+
+
+   .. list-table:: Methods
+      :header-rows: 0
+      :widths: auto
+      :class: summarytable
+
+      * - :py:obj:`run_pass <numba_dpex.core.parfors.parfor_pass.ParforPass.run_pass>`\ (state)
+        - Convert data-parallel computations into Parfor nodes.
+
+
+   .. rubric:: Members
+
+   .. py:method:: run_pass(state)
+
+      Convert data-parallel computations into Parfor nodes.
+
+      Exactly same as the original one, but with mock to _ParforPass.
+
+
+
+
+Functions
+---------
+.. py:function:: get_dpnp_ufunc_typ(func)
+
+   get type of the incoming function from builtin registry
+
+
+
+
+

dev/genindex.html

@@ -394,6 +394,8 @@ <h2>C</h2>
         <li><a href="autoapi/numba_dpex/kernel_api/index.html#numba_dpex.kernel_api.AddressSpace.CONSTANT">CONSTANT (AddressSpace attribute)</a>, <a href="autoapi/numba_dpex/kernel_api/memory_enums/index.html#numba_dpex.kernel_api.memory_enums.AddressSpace.CONSTANT">[1]</a>
 </li>
         <li><a href="autoapi/numba_dpex/kernel_api/index.html#numba_dpex.kernel_api.MemoryOrder.CONSUME_UNSUPPORTED">CONSUME_UNSUPPORTED (MemoryOrder attribute)</a>, <a href="autoapi/numba_dpex/kernel_api/memory_enums/index.html#numba_dpex.kernel_api.memory_enums.MemoryOrder.CONSUME_UNSUPPORTED">[1]</a>
+</li>
+        <li><a href="autoapi/numba_dpex/core/parfors/parfor_pass/index.html#numba_dpex.core.parfors.parfor_pass.ConvertDPNPPass">ConvertDPNPPass (class in numba_dpex.core.parfors.parfor_pass)</a>
 </li>
         <li><a href="autoapi/numba_dpex/core/index.html#numba_dpex.core.Array.copy">copy() (Array method)</a>, <a href="autoapi/numba_dpex/core/types/array_type/index.html#numba_dpex.core.types.array_type.Array.copy">[1]</a>, <a href="autoapi/numba_dpex/core/types/index.html#numba_dpex.core.types.Array.copy">[2]</a>
 
@@ -667,6 +669,8 @@ <h2>G</h2>
         <li><a href="autoapi/numba_dpex/core/datamodel/models/index.html#numba_dpex.core.datamodel.models.GenericPointerModel">GenericPointerModel (class in numba_dpex.core.datamodel.models)</a>
 </li>
         <li><a href="autoapi/numba_dpex/core/boxing/index.html#id4">get() (Range method)</a>, <a href="autoapi/numba_dpex/core/boxing/index.html#numba_dpex.core.boxing.Range.get">[1]</a>, <a href="autoapi/numba_dpex/kernel_api/index.html#numba_dpex.kernel_api.Range.get">[2]</a>, <a href="autoapi/numba_dpex/kernel_api/ranges/index.html#numba_dpex.kernel_api.ranges.Range.get">[3]</a>
+</li>
+        <li><a href="autoapi/numba_dpex/core/parfors/parfor_pass/index.html#numba_dpex.core.parfors.parfor_pass.get_dpnp_ufunc_typ">get_dpnp_ufunc_typ() (in module numba_dpex.core.parfors.parfor_pass)</a>
 </li>
         <li><a href="autoapi/numba_dpex/core/datamodel/models/index.html#numba_dpex.core.datamodel.models.get_flattened_member_count">get_flattened_member_count() (in module numba_dpex.core.datamodel.models)</a>
 </li>
@@ -1021,6 +1025,8 @@ <h2>M</h2>
           <li><a href="autoapi/numba_dpex/core/parfors/kernel_templates/reduction_template/index.html#module-numba_dpex.core.parfors.kernel_templates.reduction_template">numba_dpex.core.parfors.kernel_templates.reduction_template</a>
 </li>
           <li><a href="autoapi/numba_dpex/core/parfors/parfor_lowerer/index.html#module-numba_dpex.core.parfors.parfor_lowerer">numba_dpex.core.parfors.parfor_lowerer</a>
+</li>
+          <li><a href="autoapi/numba_dpex/core/parfors/parfor_pass/index.html#module-numba_dpex.core.parfors.parfor_pass">numba_dpex.core.parfors.parfor_pass</a>
 </li>
           <li><a href="autoapi/numba_dpex/core/parfors/reduction_helper/index.html#module-numba_dpex.core.parfors.reduction_helper">numba_dpex.core.parfors.reduction_helper</a>
 </li>
@@ -1343,6 +1349,13 @@ <h2>N</h2>
 
         <ul>
           <li><a href="autoapi/numba_dpex/core/parfors/parfor_lowerer/index.html#module-numba_dpex.core.parfors.parfor_lowerer">module</a>
+</li>
+        </ul></li>
+        <li>
+    numba_dpex.core.parfors.parfor_pass
+
+        <ul>
+          <li><a href="autoapi/numba_dpex/core/parfors/parfor_pass/index.html#module-numba_dpex.core.parfors.parfor_pass">module</a>
 </li>
         </ul></li>
         <li>
@@ -1373,15 +1386,15 @@ <h2>N</h2>
           <li><a href="autoapi/numba_dpex/core/passes/dufunc_inliner/index.html#module-numba_dpex.core.passes.dufunc_inliner">module</a>
 </li>
         </ul></li>
+    </ul></td>
+    <td style="width: 33%; vertical-align: top;"><ul>
         <li>
     numba_dpex.core.passes.parfor_legalize_cfd_pass
 
         <ul>
           <li><a href="autoapi/numba_dpex/core/passes/parfor_legalize_cfd_pass/index.html#module-numba_dpex.core.passes.parfor_legalize_cfd_pass">module</a>
 </li>
         </ul></li>
-    </ul></td>
-    <td style="width: 33%; vertical-align: top;"><ul>
         <li>
     numba_dpex.core.passes.passes
 
@@ -1693,6 +1706,8 @@ <h2>P</h2>
         <li><a href="autoapi/numba_dpex/core/parfors/parfor_lowerer/index.html#numba_dpex.core.parfors.parfor_lowerer.ParforLowerFactory">ParforLowerFactory (class in numba_dpex.core.parfors.parfor_lowerer)</a>
 </li>
         <li><a href="autoapi/numba_dpex/core/parfors/parfor_lowerer/index.html#numba_dpex.core.parfors.parfor_lowerer.ParforLowerImpl">ParforLowerImpl (class in numba_dpex.core.parfors.parfor_lowerer)</a>
+</li>
+        <li><a href="autoapi/numba_dpex/core/parfors/parfor_pass/index.html#numba_dpex.core.parfors.parfor_pass.ParforPass">ParforPass (class in numba_dpex.core.parfors.parfor_pass)</a>
 </li>
     </ul></td>
     <td style="width: 33%; vertical-align: top;"><ul>
@@ -1762,10 +1777,10 @@ <h2>R</h2>
         </ul></li>
         <li><a href="autoapi/numba_dpex/core/parfors/reduction_helper/index.html#numba_dpex.core.parfors.reduction_helper.ReductionHelper">ReductionHelper (class in numba_dpex.core.parfors.reduction_helper)</a>
 </li>
-    </ul></td>
-    <td style="width: 33%; vertical-align: top;"><ul>
         <li><a href="autoapi/numba_dpex/core/parfors/reduction_helper/index.html#numba_dpex.core.parfors.reduction_helper.ReductionKernelVariables">ReductionKernelVariables (class in numba_dpex.core.parfors.reduction_helper)</a>
 </li>
+    </ul></td>
+    <td style="width: 33%; vertical-align: top;"><ul>
         <li><a href="autoapi/numba_dpex/core/lowering/index.html#numba_dpex.core.lowering.registry">registry (in module numba_dpex.core.lowering)</a>
 
         <ul>
@@ -1788,6 +1803,8 @@ <h2>R</h2>
           <li><a href="autoapi/numba_dpex/core/passes/index.html#numba_dpex.core.passes.NoPythonBackend.run_pass">(NoPythonBackend method)</a>, <a href="autoapi/numba_dpex/core/passes/passes/index.html#numba_dpex.core.passes.passes.NoPythonBackend.run_pass">[1]</a>
 </li>
           <li><a href="autoapi/numba_dpex/core/passes/index.html#numba_dpex.core.passes.ParforLegalizeCFDPass.run_pass">(ParforLegalizeCFDPass method)</a>, <a href="autoapi/numba_dpex/core/passes/parfor_legalize_cfd_pass/index.html#numba_dpex.core.passes.parfor_legalize_cfd_pass.ParforLegalizeCFDPass.run_pass">[1]</a>
+</li>
+          <li><a href="autoapi/numba_dpex/core/parfors/parfor_pass/index.html#numba_dpex.core.parfors.parfor_pass.ParforPass.run_pass">(ParforPass method)</a>
 </li>
           <li><a href="autoapi/numba_dpex/core/passes/passes/index.html#numba_dpex.core.passes.passes.QualNameDisambiguationLowering.run_pass">(QualNameDisambiguationLowering method)</a>
 </li>

dev/py-modindex.html

@@ -411,6 +411,12 @@ <h1>Python Module Index</h1>
         <a href="autoapi/numba_dpex/core/parfors/parfor_lowerer/index.html#module-numba_dpex.core.parfors.parfor_lowerer"><code class="xref">numba_dpex.core.parfors.parfor_lowerer</code></a></td><td>
     <em></em></td>
   </tr>
+  <tr class="cg-1">
+    <td></td>
+    <td>&#160;&#160;&#160;
+        <a href="autoapi/numba_dpex/core/parfors/parfor_pass/index.html#module-numba_dpex.core.parfors.parfor_pass"><code class="xref">numba_dpex.core.parfors.parfor_pass</code></a></td><td>
+    <em></em></td>
+  </tr>
   <tr class="cg-1">
     <td></td>
     <td>&#160;&#160;&#160;