|
14 | 14 | </script>
|
15 | 15 |
|
16 | 16 | <meta name="viewport" content="width=device-width, initial-scale=1.0" />
|
17 |
| - <title>dpnp.dpnp_array — Data Parallel Extension for NumPy 0.19.0dev1+9.g463af2275b8 documentation</title> |
| 17 | + <title>dpnp.dpnp_array — Data Parallel Extension for NumPy 0.19.0dev1+10.ge2c04695cd4 documentation</title> |
18 | 18 | <link rel="stylesheet" type="text/css" href="../../_static/pygments.css?v=03e43079" />
|
19 | 19 | <link rel="stylesheet" type="text/css" href="../../_static/css/theme.css?v=e59714d7" />
|
20 | 20 |
|
21 | 21 |
|
22 | 22 | <script src="../../_static/jquery.js?v=5d32c60e"></script>
|
23 | 23 | <script src="../../_static/_sphinx_javascript_frameworks_compat.js?v=2cd50e6c"></script>
|
24 |
| - <script src="../../_static/documentation_options.js?v=378a2deb"></script> |
| 24 | + <script src="../../_static/documentation_options.js?v=c2e0af1a"></script> |
25 | 25 | <script src="../../_static/doctools.js?v=9bcbadda"></script>
|
26 | 26 | <script src="../../_static/sphinx_highlight.js?v=dc90522c"></script>
|
27 | 27 | <script src="../../_static/js/theme.js"></script>
|
@@ -114,6 +114,7 @@ <h1>Source code for dpnp.dpnp_array</h1><div class="highlight"><pre>
|
114 | 114 | <span class="kn">from</span><span class="w"> </span><span class="nn">dpctl.tensor._numpy_helper</span><span class="w"> </span><span class="kn">import</span> <span class="n">AxisError</span>
|
115 | 115 |
|
116 | 116 | <span class="kn">import</span><span class="w"> </span><span class="nn">dpnp</span>
|
| 117 | +<span class="kn">import</span><span class="w"> </span><span class="nn">dpnp.memory</span><span class="w"> </span><span class="k">as</span><span class="w"> </span><span class="nn">dpm</span> |
117 | 118 |
|
118 | 119 |
|
119 | 120 | <span class="k">def</span><span class="w"> </span><span class="nf">_get_unwrapped_index_key</span><span class="p">(</span><span class="n">key</span><span class="p">):</span>
|
@@ -163,9 +164,12 @@ <h1>Source code for dpnp.dpnp_array</h1><div class="highlight"><pre>
|
163 | 164 | <span class="n">order</span> <span class="o">=</span> <span class="s2">"C"</span>
|
164 | 165 |
|
165 | 166 | <span class="k">if</span> <span class="n">buffer</span> <span class="ow">is</span> <span class="ow">not</span> <span class="kc">None</span><span class="p">:</span>
|
166 |
| - <span class="n">buffer</span> <span class="o">=</span> <span class="n">dpnp</span><span class="o">.</span><span class="n">get_usm_ndarray</span><span class="p">(</span><span class="n">buffer</span><span class="p">)</span> |
| 167 | + <span class="c1"># expecting to have buffer as dpnp.ndarray and usm_ndarray,</span> |
| 168 | + <span class="c1"># or as USM memory allocation</span> |
| 169 | + <span class="k">if</span> <span class="nb">isinstance</span><span class="p">(</span><span class="n">buffer</span><span class="p">,</span> <span class="n">dpnp_array</span><span class="p">):</span> |
| 170 | + <span class="n">buffer</span> <span class="o">=</span> <span class="n">buffer</span><span class="o">.</span><span class="n">get_array</span><span class="p">()</span> |
167 | 171 |
|
168 |
| - <span class="k">if</span> <span class="n">dtype</span> <span class="ow">is</span> <span class="kc">None</span><span class="p">:</span> |
| 172 | + <span class="k">if</span> <span class="n">dtype</span> <span class="ow">is</span> <span class="kc">None</span> <span class="ow">and</span> <span class="nb">hasattr</span><span class="p">(</span><span class="n">buffer</span><span class="p">,</span> <span class="s2">"dtype"</span><span class="p">):</span> |
169 | 173 | <span class="n">dtype</span> <span class="o">=</span> <span class="n">buffer</span><span class="o">.</span><span class="n">dtype</span>
|
170 | 174 | <span class="k">else</span><span class="p">:</span>
|
171 | 175 | <span class="n">buffer</span> <span class="o">=</span> <span class="n">usm_type</span>
|
@@ -1150,7 +1154,15 @@ <h1>Source code for dpnp.dpnp_array</h1><div class="highlight"><pre>
|
1150 | 1154 | <span class="k">return</span> <span class="n">dpnp</span><span class="o">.</span><span class="n">cumsum</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></div>
|
1151 | 1155 |
|
1152 | 1156 |
|
1153 |
| - <span class="c1"># 'data',</span> |
| 1157 | + <span class="nd">@property</span> |
| 1158 | + <span class="k">def</span><span class="w"> </span><span class="nf">data</span><span class="p">(</span><span class="bp">self</span><span class="p">):</span> |
| 1159 | +<span class="w"> </span><span class="sd">"""</span> |
| 1160 | +<span class="sd"> Python object pointing to the start of USM memory allocation with the</span> |
| 1161 | +<span class="sd"> array's data.</span> |
| 1162 | + |
| 1163 | +<span class="sd"> """</span> |
| 1164 | + |
| 1165 | + <span class="k">return</span> <span class="n">dpm</span><span class="o">.</span><span class="n">create_data</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">_array_obj</span><span class="p">)</span> |
1154 | 1166 |
|
1155 | 1167 | <div class="viewcode-block" id="dpnp_array.diagonal">
|
1156 | 1168 | <a class="viewcode-back" href="../../reference/generated/dpnp.dpnp_array.dpnp_array.diagonal.html#dpnp.dpnp_array.dpnp_array.diagonal">[docs]</a>
|
|
0 commit comments