|
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.17.0dev4+4.g952a798fe8 documentation</title> |
| 17 | + <title>dpnp.dpnp_array — Data Parallel Extension for NumPy 0.17.0dev4+5.g6cc23484bc9 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=aa6b5ff6"></script> |
| 24 | + <script src="../../_static/documentation_options.js?v=6db27343"></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>
|
@@ -243,16 +243,6 @@ <h1>Source code for dpnp.dpnp_array</h1><div class="highlight"><pre>
|
243 | 243 |
|
244 | 244 | <span class="k">return</span> <span class="n">dpnp_array</span><span class="o">.</span><span class="n">_create_from_usm_ndarray</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">_array_obj</span><span class="o">.</span><span class="n">mT</span><span class="p">)</span>
|
245 | 245 |
|
246 |
| -<div class="viewcode-block" id="dpnp_array.to_device"> |
247 |
| -<a class="viewcode-back" href="../../reference/generated/dpnp.dpnp_array.dpnp_array.to_device.html#dpnp.dpnp_array.dpnp_array.to_device">[docs]</a> |
248 |
| - <span class="k">def</span><span class="w"> </span><span class="nf">to_device</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">target_device</span><span class="p">):</span> |
249 |
| -<span class="w"> </span><span class="sd">"""Transfer array to target device."""</span> |
250 |
| - |
251 |
| - <span class="k">return</span> <span class="n">dpnp_array</span><span class="p">(</span> |
252 |
| - <span class="n">shape</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">shape</span><span class="p">,</span> <span class="n">buffer</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">get_array</span><span class="p">()</span><span class="o">.</span><span class="n">to_device</span><span class="p">(</span><span class="n">target_device</span><span class="p">)</span> |
253 |
| - <span class="p">)</span></div> |
254 |
| - |
255 |
| - |
256 | 246 | <span class="nd">@property</span>
|
257 | 247 | <span class="k">def</span><span class="w"> </span><span class="nf">sycl_queue</span><span class="p">(</span><span class="bp">self</span><span class="p">):</span>
|
258 | 248 | <span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">_array_obj</span><span class="o">.</span><span class="n">sycl_queue</span>
|
@@ -1900,6 +1890,51 @@ <h1>Source code for dpnp.dpnp_array</h1><div class="highlight"><pre>
|
1900 | 1890 | <span class="k">return</span> <span class="n">dpnp</span><span class="o">.</span><span class="n">take</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">indices</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">mode</span><span class="o">=</span><span class="n">mode</span><span class="p">)</span></div>
|
1901 | 1891 |
|
1902 | 1892 |
|
| 1893 | +<div class="viewcode-block" id="dpnp_array.to_device"> |
| 1894 | +<a class="viewcode-back" href="../../reference/generated/dpnp.dpnp_array.dpnp_array.to_device.html#dpnp.dpnp_array.dpnp_array.to_device">[docs]</a> |
| 1895 | + <span class="k">def</span><span class="w"> </span><span class="nf">to_device</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">device</span><span class="p">,</span> <span class="o">/</span><span class="p">,</span> <span class="o">*</span><span class="p">,</span> <span class="n">stream</span><span class="o">=</span><span class="kc">None</span><span class="p">):</span> |
| 1896 | +<span class="w"> </span><span class="sd">"""</span> |
| 1897 | +<span class="sd"> Transfers this array to specified target device.</span> |
| 1898 | + |
| 1899 | +<span class="sd"> Parameters</span> |
| 1900 | +<span class="sd"> ----------</span> |
| 1901 | +<span class="sd"> device : {string, SyclDevice, SyclQueue}</span> |
| 1902 | +<span class="sd"> Array API concept of target device. It can be an OneAPI filter</span> |
| 1903 | +<span class="sd"> selector string, an instance of :class:`dpctl.SyclDevice`</span> |
| 1904 | +<span class="sd"> corresponding to a non-partitioned SYCL device, an instance of</span> |
| 1905 | +<span class="sd"> :class:`dpctl.SyclQueue`, or a :class:`dpctl.tensor.Device` object</span> |
| 1906 | +<span class="sd"> returned by :obj:`dpnp.dpnp_array.dpnp_array.device` property.</span> |
| 1907 | +<span class="sd"> stream : {SyclQueue, None}, optional</span> |
| 1908 | +<span class="sd"> Execution queue to synchronize with. If ``None``, synchronization</span> |
| 1909 | +<span class="sd"> is not performed.</span> |
| 1910 | +<span class="sd"> Default: ``None``.</span> |
| 1911 | + |
| 1912 | +<span class="sd"> Returns</span> |
| 1913 | +<span class="sd"> -------</span> |
| 1914 | +<span class="sd"> out : dpnp.ndarray</span> |
| 1915 | +<span class="sd"> A view if data copy is not required, and a copy otherwise.</span> |
| 1916 | +<span class="sd"> If copying is required, it is done by copying from the original</span> |
| 1917 | +<span class="sd"> allocation device to the host, followed by copying from host</span> |
| 1918 | +<span class="sd"> to the target device.</span> |
| 1919 | + |
| 1920 | +<span class="sd"> Examples</span> |
| 1921 | +<span class="sd"> --------</span> |
| 1922 | +<span class="sd"> >>> import dpnp as np, dpctl</span> |
| 1923 | +<span class="sd"> >>> x = np.full(100, 2, dtype=np.int64)</span> |
| 1924 | +<span class="sd"> >>> q_prof = dpctl.SyclQueue(x.sycl_device, property="enable_profiling")</span> |
| 1925 | +<span class="sd"> >>> # return a view with profile-enabled queue</span> |
| 1926 | +<span class="sd"> >>> y = x.to_device(q_prof)</span> |
| 1927 | +<span class="sd"> >>> timer = dpctl.SyclTimer()</span> |
| 1928 | +<span class="sd"> >>> with timer(q_prof):</span> |
| 1929 | +<span class="sd"> ... z = y * y</span> |
| 1930 | +<span class="sd"> >>> print(timer.dt)</span> |
| 1931 | + |
| 1932 | +<span class="sd"> """</span> |
| 1933 | + |
| 1934 | + <span class="n">usm_res</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">_array_obj</span><span class="o">.</span><span class="n">to_device</span><span class="p">(</span><span class="n">device</span><span class="p">,</span> <span class="n">stream</span><span class="o">=</span><span class="n">stream</span><span class="p">)</span> |
| 1935 | + <span class="k">return</span> <span class="n">dpnp_array</span><span class="o">.</span><span class="n">_create_from_usm_ndarray</span><span class="p">(</span><span class="n">usm_res</span><span class="p">)</span></div> |
| 1936 | + |
| 1937 | + |
1903 | 1938 | <span class="c1"># 'tobytes',</span>
|
1904 | 1939 | <span class="c1"># 'tofile',</span>
|
1905 | 1940 | <span class="c1"># 'tolist',</span>
|
|
0 commit comments