Skip to content

Commit

Permalink
Leverage on out keyword in reduction and where functions of `dpct…
Browse files Browse the repository at this point in the history
…l.tensor` (IntelPython#1808)

* Pass out keyword to dpctl in reduction functions

* Pass out and order keywords to dpctl.tensor.where

* Add wrapping methods to reduce code duplication

* Update docstring comments

* Resolve issue from spelling check

* Separate wrapping function to reduction utils

* Fix pre-commit issues

* Fixed a typo

* Applied review comments 4dc16b4
  • Loading branch information
github-actions[bot] committed May 6, 2024
1 parent c48d510 commit 032874d
Show file tree
Hide file tree
Showing 486 changed files with 2,369 additions and 2,128 deletions.
2 changes: 1 addition & 1 deletion .buildinfo
Original file line number Diff line number Diff line change
@@ -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: bb50a0e5db9acc7a1dc0553abfd6d09b
config: 9d7aab329f47c7fe535884a4127631a5
tags: 645f666f9bcd5a90fca523b33c5a78b7
50 changes: 37 additions & 13 deletions _modules/dpnp/dpnp_array.html
Original file line number Diff line number Diff line change
Expand Up @@ -12,7 +12,7 @@
</script>

<meta name="viewport" content="width=device-width, initial-scale=1.0" />
<title>dpnp.dpnp_array &mdash; Data Parallel Extension for NumPy 0.11.1+560.g36ae1febb documentation</title>
<title>dpnp.dpnp_array &mdash; Data Parallel Extension for NumPy 0.11.1+561.g4dc16b489 documentation</title>
<link rel="stylesheet" type="text/css" href="../../_static/pygments.css?v=fa44fd50" />
<link rel="stylesheet" type="text/css" href="../../_static/css/theme.css?v=19f00094" />

Expand All @@ -23,7 +23,7 @@

<script src="../../_static/jquery.js?v=5d32c60e"></script>
<script src="../../_static/_sphinx_javascript_frameworks_compat.js?v=2cd50e6c"></script>
<script src="../../_static/documentation_options.js?v=9633119b"></script>
<script src="../../_static/documentation_options.js?v=d526ccd8"></script>
<script src="../../_static/doctools.js?v=9a2dae69"></script>
<script src="../../_static/sphinx_highlight.js?v=dc90522c"></script>
<script src="../../_static/js/theme.js"></script>
Expand All @@ -43,7 +43,7 @@
Data Parallel Extension for NumPy
</a>
<div class="version">
0.11.1+560.g36ae1febb
0.11.1+561.g4dc16b489
</div>
<div role="search">
<form id="rtd-search-form" class="wy-form" action="../../search.html" method="get">
Expand Down Expand Up @@ -658,7 +658,8 @@ <h1>Source code for dpnp.dpnp_array</h1><div class="highlight"><pre>
<span class="sd"> Refer to :obj:`dpnp.argmax` for full documentation.</span>

<span class="sd"> &quot;&quot;&quot;</span>
<span class="k">return</span> <span class="n">dpnp</span><span class="o">.</span><span class="n">argmax</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">axis</span><span class="p">,</span> <span class="n">out</span><span class="p">,</span> <span class="n">keepdims</span><span class="o">=</span><span class="n">keepdims</span><span class="p">)</span></div>

<span class="k">return</span> <span class="n">dpnp</span><span class="o">.</span><span class="n">argmax</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">axis</span><span class="o">=</span><span class="n">axis</span><span class="p">,</span> <span class="n">out</span><span class="o">=</span><span class="n">out</span><span class="p">,</span> <span class="n">keepdims</span><span class="o">=</span><span class="n">keepdims</span><span class="p">)</span></div>


<div class="viewcode-block" id="dpnp_array.argmin">
Expand All @@ -670,7 +671,8 @@ <h1>Source code for dpnp.dpnp_array</h1><div class="highlight"><pre>
<span class="sd"> Refer to :obj:`dpnp.argmin` for full documentation.</span>

<span class="sd"> &quot;&quot;&quot;</span>
<span class="k">return</span> <span class="n">dpnp</span><span class="o">.</span><span class="n">argmin</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">axis</span><span class="p">,</span> <span class="n">out</span><span class="p">,</span> <span class="n">keepdims</span><span class="o">=</span><span class="n">keepdims</span><span class="p">)</span></div>

<span class="k">return</span> <span class="n">dpnp</span><span class="o">.</span><span class="n">argmin</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">axis</span><span class="o">=</span><span class="n">axis</span><span class="p">,</span> <span class="n">out</span><span class="o">=</span><span class="n">out</span><span class="p">,</span> <span class="n">keepdims</span><span class="o">=</span><span class="n">keepdims</span><span class="p">)</span></div>


<span class="c1"># &#39;argpartition&#39;,</span>
Expand Down Expand Up @@ -727,13 +729,15 @@ <h1>Source code for dpnp.dpnp_array</h1><div class="highlight"><pre>
<span class="sd"> casting : {&#39;no&#39;, &#39;equiv&#39;, &#39;safe&#39;, &#39;same_kind&#39;, &#39;unsafe&#39;}, optional</span>
<span class="sd"> Controls what kind of data casting may occur.</span>
<span class="sd"> Defaults to ``&#39;unsafe&#39;`` for backwards compatibility.</span>

<span class="sd"> - &#39;no&#39; means the data types should not be cast at all.</span>
<span class="sd"> - &#39;equiv&#39; means only byte-order changes are allowed.</span>
<span class="sd"> - &#39;safe&#39; means only casts which can preserve values are allowed.</span>
<span class="sd"> - &#39;same_kind&#39; means only safe casts or casts within a kind, like</span>
<span class="sd"> float64 to float32, are allowed.</span>
<span class="sd"> - &#39;unsafe&#39; means any data conversions may be done.</span>
<span class="sd"> copy : bool, optional</span>

<span class="sd"> copy : {bool}, optional</span>
<span class="sd"> By default, ``astype`` always returns a newly allocated array. If</span>
<span class="sd"> this is set to ``False``, and the `dtype`, `order`, and `subok`</span>
<span class="sd"> requirements are satisfied, the input array is returned instead of</span>
Expand Down Expand Up @@ -1132,7 +1136,14 @@ <h1>Source code for dpnp.dpnp_array</h1><div class="highlight"><pre>

<span class="sd"> &quot;&quot;&quot;</span>

<span class="k">return</span> <span class="n">dpnp</span><span class="o">.</span><span class="n">max</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">axis</span><span class="p">,</span> <span class="n">out</span><span class="p">,</span> <span class="n">keepdims</span><span class="p">,</span> <span class="n">initial</span><span class="p">,</span> <span class="n">where</span><span class="p">)</span></div>
<span class="k">return</span> <span class="n">dpnp</span><span class="o">.</span><span class="n">max</span><span class="p">(</span>
<span class="bp">self</span><span class="p">,</span>
<span class="n">axis</span><span class="o">=</span><span class="n">axis</span><span class="p">,</span>
<span class="n">out</span><span class="o">=</span><span class="n">out</span><span class="p">,</span>
<span class="n">keepdims</span><span class="o">=</span><span class="n">keepdims</span><span class="p">,</span>
<span class="n">initial</span><span class="o">=</span><span class="n">initial</span><span class="p">,</span>
<span class="n">where</span><span class="o">=</span><span class="n">where</span><span class="p">,</span>
<span class="p">)</span></div>


<div class="viewcode-block" id="dpnp_array.mean">
Expand Down Expand Up @@ -1167,7 +1178,14 @@ <h1>Source code for dpnp.dpnp_array</h1><div class="highlight"><pre>

<span class="sd"> &quot;&quot;&quot;</span>

<span class="k">return</span> <span class="n">dpnp</span><span class="o">.</span><span class="n">min</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">axis</span><span class="p">,</span> <span class="n">out</span><span class="p">,</span> <span class="n">keepdims</span><span class="p">,</span> <span class="n">initial</span><span class="p">,</span> <span class="n">where</span><span class="p">)</span></div>
<span class="k">return</span> <span class="n">dpnp</span><span class="o">.</span><span class="n">min</span><span class="p">(</span>
<span class="bp">self</span><span class="p">,</span>
<span class="n">axis</span><span class="o">=</span><span class="n">axis</span><span class="p">,</span>
<span class="n">out</span><span class="o">=</span><span class="n">out</span><span class="p">,</span>
<span class="n">keepdims</span><span class="o">=</span><span class="n">keepdims</span><span class="p">,</span>
<span class="n">initial</span><span class="o">=</span><span class="n">initial</span><span class="p">,</span>
<span class="n">where</span><span class="o">=</span><span class="n">where</span><span class="p">,</span>
<span class="p">)</span></div>


<span class="nd">@property</span>
Expand Down Expand Up @@ -1250,7 +1268,15 @@ <h1>Source code for dpnp.dpnp_array</h1><div class="highlight"><pre>

<span class="sd"> &quot;&quot;&quot;</span>

<span class="k">return</span> <span class="n">dpnp</span><span class="o">.</span><span class="n">prod</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">axis</span><span class="p">,</span> <span class="n">dtype</span><span class="p">,</span> <span class="n">out</span><span class="p">,</span> <span class="n">keepdims</span><span class="p">,</span> <span class="n">initial</span><span class="p">,</span> <span class="n">where</span><span class="p">)</span></div>
<span class="k">return</span> <span class="n">dpnp</span><span class="o">.</span><span class="n">prod</span><span class="p">(</span>
<span class="bp">self</span><span class="p">,</span>
<span class="n">axis</span><span class="o">=</span><span class="n">axis</span><span class="p">,</span>
<span class="n">dtype</span><span class="o">=</span><span class="n">dtype</span><span class="p">,</span>
<span class="n">out</span><span class="o">=</span><span class="n">out</span><span class="p">,</span>
<span class="n">keepdims</span><span class="o">=</span><span class="n">keepdims</span><span class="p">,</span>
<span class="n">initial</span><span class="o">=</span><span class="n">initial</span><span class="p">,</span>
<span class="n">where</span><span class="o">=</span><span class="n">where</span><span class="p">,</span>
<span class="p">)</span></div>


<div class="viewcode-block" id="dpnp_array.put">
Expand Down Expand Up @@ -1521,13 +1547,11 @@ <h1>Source code for dpnp.dpnp_array</h1><div class="highlight"><pre>
<a class="viewcode-back" href="../../reference/generated/dpnp.dpnp_array.dpnp_array.html#dpnp.dpnp_array.dpnp_array.sum">[docs]</a>
<span class="k">def</span> <span class="nf">sum</span><span class="p">(</span>
<span class="bp">self</span><span class="p">,</span>
<span class="o">/</span><span class="p">,</span>
<span class="o">*</span><span class="p">,</span>
<span class="n">axis</span><span class="o">=</span><span class="kc">None</span><span class="p">,</span>
<span class="n">dtype</span><span class="o">=</span><span class="kc">None</span><span class="p">,</span>
<span class="n">keepdims</span><span class="o">=</span><span class="kc">False</span><span class="p">,</span>
<span class="n">out</span><span class="o">=</span><span class="kc">None</span><span class="p">,</span>
<span class="n">initial</span><span class="o">=</span><span class="mi">0</span><span class="p">,</span>
<span class="n">keepdims</span><span class="o">=</span><span class="kc">False</span><span class="p">,</span>
<span class="n">initial</span><span class="o">=</span><span class="kc">None</span><span class="p">,</span>
<span class="n">where</span><span class="o">=</span><span class="kc">True</span><span class="p">,</span>
<span class="p">):</span>
<span class="w"> </span><span class="sd">&quot;&quot;&quot;</span>
Expand Down
6 changes: 3 additions & 3 deletions _modules/dpnp/dpnp_flatiter.html
Original file line number Diff line number Diff line change
Expand Up @@ -12,7 +12,7 @@
</script>

<meta name="viewport" content="width=device-width, initial-scale=1.0" />
<title>dpnp.dpnp_flatiter &mdash; Data Parallel Extension for NumPy 0.11.1+560.g36ae1febb documentation</title>
<title>dpnp.dpnp_flatiter &mdash; Data Parallel Extension for NumPy 0.11.1+561.g4dc16b489 documentation</title>
<link rel="stylesheet" type="text/css" href="../../_static/pygments.css?v=fa44fd50" />
<link rel="stylesheet" type="text/css" href="../../_static/css/theme.css?v=19f00094" />

Expand All @@ -23,7 +23,7 @@

<script src="../../_static/jquery.js?v=5d32c60e"></script>
<script src="../../_static/_sphinx_javascript_frameworks_compat.js?v=2cd50e6c"></script>
<script src="../../_static/documentation_options.js?v=9633119b"></script>
<script src="../../_static/documentation_options.js?v=d526ccd8"></script>
<script src="../../_static/doctools.js?v=9a2dae69"></script>
<script src="../../_static/sphinx_highlight.js?v=dc90522c"></script>
<script src="../../_static/js/theme.js"></script>
Expand All @@ -43,7 +43,7 @@
Data Parallel Extension for NumPy
</a>
<div class="version">
0.11.1+560.g36ae1febb
0.11.1+561.g4dc16b489
</div>
<div role="search">
<form id="rtd-search-form" class="wy-form" action="../../search.html" method="get">
Expand Down
8 changes: 4 additions & 4 deletions _modules/dpnp/dpnp_iface.html
Original file line number Diff line number Diff line change
Expand Up @@ -12,7 +12,7 @@
</script>

<meta name="viewport" content="width=device-width, initial-scale=1.0" />
<title>dpnp.dpnp_iface &mdash; Data Parallel Extension for NumPy 0.11.1+560.g36ae1febb documentation</title>
<title>dpnp.dpnp_iface &mdash; Data Parallel Extension for NumPy 0.11.1+561.g4dc16b489 documentation</title>
<link rel="stylesheet" type="text/css" href="../../_static/pygments.css?v=fa44fd50" />
<link rel="stylesheet" type="text/css" href="../../_static/css/theme.css?v=19f00094" />

Expand All @@ -23,7 +23,7 @@

<script src="../../_static/jquery.js?v=5d32c60e"></script>
<script src="../../_static/_sphinx_javascript_frameworks_compat.js?v=2cd50e6c"></script>
<script src="../../_static/documentation_options.js?v=9633119b"></script>
<script src="../../_static/documentation_options.js?v=d526ccd8"></script>
<script src="../../_static/doctools.js?v=9a2dae69"></script>
<script src="../../_static/sphinx_highlight.js?v=dc90522c"></script>
<script src="../../_static/js/theme.js"></script>
Expand All @@ -43,7 +43,7 @@
Data Parallel Extension for NumPy
</a>
<div class="version">
0.11.1+560.g36ae1febb
0.11.1+561.g4dc16b489
</div>
<div role="search">
<form id="rtd-search-form" class="wy-form" action="../../search.html" method="get">
Expand Down Expand Up @@ -307,7 +307,7 @@ <h1>Source code for dpnp.dpnp_iface</h1><div class="highlight"><pre>
<span class="sd"> float64 to float32, are allowed.</span>
<span class="sd"> - &#39;unsafe&#39; means any data conversions may be done.</span>

<span class="sd"> copy : bool, optional</span>
<span class="sd"> copy : {bool}, optional</span>
<span class="sd"> By default, ``astype`` always returns a newly allocated array. If this</span>
<span class="sd"> is set to ``False``, and the `dtype`, `order`, and `subok` requirements</span>
<span class="sd"> are satisfied, the input array is returned instead of a copy.</span>
Expand Down
Loading

0 comments on commit 032874d

Please sign in to comment.