Merge pull request #1377 from IntelPython/fix/private_array

Fix/private array 200bf8d
IntelPython · Mar 15, 2024 · 0941992 · 0941992
1 parent cb17a31
commit 0941992
Show file tree

Hide file tree

Showing 13 changed files with 426 additions and 24 deletions.
diff --git a/dev/.buildinfo b/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: 367ebadfc19e391f77c6fb93107aa4f4
+config: 56821679db103af3637ee77a1e427291
 tags: 645f666f9bcd5a90fca523b33c5a78b7
diff --git a/...perimental/_kernel_dpcpp_spirv_overloads/_private_array_overloads/index.rst.txt b/...perimental/_kernel_dpcpp_spirv_overloads/_private_array_overloads/index.rst.txt
@@ -21,17 +21,19 @@ Overview
    :widths: auto
    :class: summarytable
 
-   * - :py:obj:`ol_private_array_ctor <numba_dpex.experimental._kernel_dpcpp_spirv_overloads._private_array_overloads.ol_private_array_ctor>`\ (shape, dtype)
-     - Overload of the constructor for the class
+   * - :py:obj:`type_interval <numba_dpex.experimental._kernel_dpcpp_spirv_overloads._private_array_overloads.type_interval>`\ (context)
+     - Sets type of the constructor for the class
+   * - :py:obj:`dpex_private_array_lower <numba_dpex.experimental._kernel_dpcpp_spirv_overloads._private_array_overloads.dpex_private_array_lower>`\ (context, builder, sig, args)
+     - Implements lower for the class:`numba_dpex.kernel_api.PrivateArray`
 
 
 
 
 Functions
 ---------
-.. py:function:: ol_private_array_ctor(shape, dtype)
+.. py:function:: type_interval(context)
 
-   Overload of the constructor for the class
+   Sets type of the constructor for the class
    class:`numba_dpex.kernel_api.PrivateArray`.
 
    :raises errors.TypingError: If the shape argument is not a shape compatible
@@ -40,6 +42,11 @@ Functions
        type.
 
 
+.. py:function:: dpex_private_array_lower(context: numba_dpex.experimental.target.DpexExpKernelTypingContext, builder: llvmlite.ir.builder.IRBuilder, sig: numba.core.typing.templates.Signature, args: list[llvmlite.ir.Value])
+
+   Implements lower for the class:`numba_dpex.kernel_api.PrivateArray`
+
+
 
 
 
diff --git a/...i/numba_dpex/experimental/_kernel_dpcpp_spirv_overloads/_registry/index.rst.txt b/...i/numba_dpex/experimental/_kernel_dpcpp_spirv_overloads/_registry/index.rst.txt
@@ -0,0 +1,32 @@
+
+:orphan:
+
+numba_dpex.experimental._kernel_dpcpp_spirv_overloads._registry
+===============================================================
+
+.. py:module:: numba_dpex.experimental._kernel_dpcpp_spirv_overloads._registry
+
+.. autoapi-nested-parse::
+
+   Implements the SPIR-V overloads for the kernel_api.PrivateArray class.
+
+
+
+Overview
+--------
+
+
+
+
+Attributes
+----------
+.. py:data:: registry
+
+   
+
+.. py:data:: lower
+
+   
+
+
+
diff --git a/dev/_sources/autoapi/numba_dpex/kernel_api/index.rst.txt b/dev/_sources/autoapi/numba_dpex/kernel_api/index.rst.txt
@@ -679,7 +679,7 @@ Classes
 
 
 
-.. py:class:: PrivateArray(shape, dtype)
+.. py:class:: PrivateArray(shape, dtype, fill_zeros=False)
 
    The ``PrivateArray`` class is an simple version of array intended to be used
    inside kernel work item.

diff --git a/dev/_sources/autoapi/numba_dpex/kernel_api/private_array/index.rst.txt b/dev/_sources/autoapi/numba_dpex/kernel_api/private_array/index.rst.txt
@@ -31,7 +31,7 @@ Overview
 Classes
 -------
 
-.. py:class:: PrivateArray(shape, dtype)
+.. py:class:: PrivateArray(shape, dtype, fill_zeros=False)
 
    The ``PrivateArray`` class is an simple version of array intended to be used
    inside kernel work item.

diff --git a/...numba_dpex/experimental/_kernel_dpcpp_spirv_overloads/_private_array_overloads/index.html b/...numba_dpex/experimental/_kernel_dpcpp_spirv_overloads/_private_array_overloads/index.html
@@ -254,8 +254,11 @@ <h2>Overview<a class="headerlink" href="#overview" title="Link to this heading">
 <table class="summarytable docutils align-default" id="id1">
 <caption><span class="caption-text">Function</span><a class="headerlink" href="#id1" title="Link to this table">#</a></caption>
 <tbody>
-<tr class="row-odd"><td><p><a class="reference internal" href="#numba_dpex.experimental._kernel_dpcpp_spirv_overloads._private_array_overloads.ol_private_array_ctor" title="numba_dpex.experimental._kernel_dpcpp_spirv_overloads._private_array_overloads.ol_private_array_ctor"><code class="xref py py-obj docutils literal notranslate"><span class="pre">ol_private_array_ctor</span></code></a>(shape, dtype)</p></td>
-<td><p>Overload of the constructor for the class</p></td>
+<tr class="row-odd"><td><p><a class="reference internal" href="#numba_dpex.experimental._kernel_dpcpp_spirv_overloads._private_array_overloads.type_interval" title="numba_dpex.experimental._kernel_dpcpp_spirv_overloads._private_array_overloads.type_interval"><code class="xref py py-obj docutils literal notranslate"><span class="pre">type_interval</span></code></a>(context)</p></td>
+<td><p>Sets type of the constructor for the class</p></td>
+</tr>
+<tr class="row-even"><td><p><a class="reference internal" href="#numba_dpex.experimental._kernel_dpcpp_spirv_overloads._private_array_overloads.dpex_private_array_lower" title="numba_dpex.experimental._kernel_dpcpp_spirv_overloads._private_array_overloads.dpex_private_array_lower"><code class="xref py py-obj docutils literal notranslate"><span class="pre">dpex_private_array_lower</span></code></a>(context, builder, sig, args)</p></td>
+<td><p>Implements lower for the class:<cite>numba_dpex.kernel_api.PrivateArray</cite></p></td>
 </tr>
 </tbody>
 </table>
@@ -264,9 +267,9 @@ <h2>Overview<a class="headerlink" href="#overview" title="Link to this heading">
 <section id="functions">
 <h2>Functions<a class="headerlink" href="#functions" title="Link to this heading">#</a></h2>
 <dl class="py function">
-<dt class="sig sig-object py" id="numba_dpex.experimental._kernel_dpcpp_spirv_overloads._private_array_overloads.ol_private_array_ctor">
-<span class="sig-name descname"><span class="pre">ol_private_array_ctor</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">shape</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">dtype</span></span></em><span class="sig-paren">)</span><a class="headerlink" href="#numba_dpex.experimental._kernel_dpcpp_spirv_overloads._private_array_overloads.ol_private_array_ctor" title="Link to this definition">#</a></dt>
-<dd><p>Overload of the constructor for the class
+<dt class="sig sig-object py" id="numba_dpex.experimental._kernel_dpcpp_spirv_overloads._private_array_overloads.type_interval">
+<span class="sig-name descname"><span class="pre">type_interval</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">context</span></span></em><span class="sig-paren">)</span><a class="headerlink" href="#numba_dpex.experimental._kernel_dpcpp_spirv_overloads._private_array_overloads.type_interval" title="Link to this definition">#</a></dt>
+<dd><p>Sets type of the constructor for the class
 class:<cite>numba_dpex.kernel_api.PrivateArray</cite>.</p>
 <dl class="field-list simple">
 <dt class="field-odd">Raises<span class="colon">:</span></dt>
@@ -280,6 +283,12 @@ <h2>Functions<a class="headerlink" href="#functions" title="Link to this heading
 </dl>
 </dd></dl>
 
+<dl class="py function">
+<dt class="sig sig-object py" id="numba_dpex.experimental._kernel_dpcpp_spirv_overloads._private_array_overloads.dpex_private_array_lower">
+<span class="sig-name descname"><span class="pre">dpex_private_array_lower</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">context</span></span><span class="p"><span class="pre">:</span></span><span class="w"> </span><span class="n"><a class="reference internal" href="../../target/index.html#numba_dpex.experimental.target.DpexExpKernelTypingContext" title="numba_dpex.experimental.target.DpexExpKernelTypingContext"><span class="pre">numba_dpex.experimental.target.DpexExpKernelTypingContext</span></a></span></em>, <em class="sig-param"><span class="n"><span class="pre">builder</span></span><span class="p"><span class="pre">:</span></span><span class="w"> </span><span class="n"><span class="pre">llvmlite.ir.builder.IRBuilder</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">sig</span></span><span class="p"><span class="pre">:</span></span><span class="w"> </span><span class="n"><span class="pre">numba.core.typing.templates.Signature</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">args</span></span><span class="p"><span class="pre">:</span></span><span class="w"> </span><span class="n"><span class="pre">list</span><span class="p"><span class="pre">[</span></span><span class="pre">llvmlite.ir.Value</span><span class="p"><span class="pre">]</span></span></span></em><span class="sig-paren">)</span><a class="headerlink" href="#numba_dpex.experimental._kernel_dpcpp_spirv_overloads._private_array_overloads.dpex_private_array_lower" title="Link to this definition">#</a></dt>
+<dd><p>Implements lower for the class:<cite>numba_dpex.kernel_api.PrivateArray</cite></p>
+</dd></dl>
+
 </section>
 </section>
 
@@ -326,7 +335,8 @@ <h2>Functions<a class="headerlink" href="#functions" title="Link to this heading
 <li><a class="reference internal" href="#">numba_dpex.experimental._kernel_dpcpp_spirv_overloads._private_array_overloads</a><ul>
 <li><a class="reference internal" href="#overview">Overview</a></li>
 <li><a class="reference internal" href="#functions">Functions</a><ul>
-<li><a class="reference internal" href="#numba_dpex.experimental._kernel_dpcpp_spirv_overloads._private_array_overloads.ol_private_array_ctor"><code class="docutils literal notranslate"><span class="pre">ol_private_array_ctor()</span></code></a></li>
+<li><a class="reference internal" href="#numba_dpex.experimental._kernel_dpcpp_spirv_overloads._private_array_overloads.type_interval"><code class="docutils literal notranslate"><span class="pre">type_interval()</span></code></a></li>
+<li><a class="reference internal" href="#numba_dpex.experimental._kernel_dpcpp_spirv_overloads._private_array_overloads.dpex_private_array_lower"><code class="docutils literal notranslate"><span class="pre">dpex_private_array_lower()</span></code></a></li>
 </ul>
 </li>
 </ul>