|
14 | 14 | </script>
|
15 | 15 |
|
16 | 16 | <meta name="viewport" content="width=device-width, initial-scale=1.0" />
|
17 |
| - <title>dpnp.dpnp_iface — Data Parallel Extension for NumPy 0.17.0dev4+3.g498e705d848 documentation</title> |
| 17 | + <title>dpnp.dpnp_iface — Data Parallel Extension for NumPy 0.17.0dev4+4.g952a798fe8 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=947258ee"></script> |
| 24 | + <script src="../../_static/documentation_options.js?v=aa6b5ff6"></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>
|
@@ -153,6 +153,7 @@ <h1>Source code for dpnp.dpnp_iface</h1><div class="highlight"><pre>
|
153 | 153 | <span class="s2">"get_result_array"</span><span class="p">,</span>
|
154 | 154 | <span class="s2">"get_usm_ndarray"</span><span class="p">,</span>
|
155 | 155 | <span class="s2">"get_usm_ndarray_or_scalar"</span><span class="p">,</span>
|
| 156 | + <span class="s2">"is_cuda_backend"</span><span class="p">,</span> |
156 | 157 | <span class="s2">"is_supported_array_or_scalar"</span><span class="p">,</span>
|
157 | 158 | <span class="s2">"is_supported_array_type"</span><span class="p">,</span>
|
158 | 159 | <span class="s2">"synchronize_array_data"</span><span class="p">,</span>
|
@@ -775,6 +776,41 @@ <h1>Source code for dpnp.dpnp_iface</h1><div class="highlight"><pre>
|
775 | 776 | <span class="k">return</span> <span class="n">a</span> <span class="k">if</span> <span class="n">dpnp</span><span class="o">.</span><span class="n">isscalar</span><span class="p">(</span><span class="n">a</span><span class="p">)</span> <span class="k">else</span> <span class="n">get_usm_ndarray</span><span class="p">(</span><span class="n">a</span><span class="p">)</span>
|
776 | 777 |
|
777 | 778 |
|
| 779 | +<span class="k">def</span><span class="w"> </span><span class="nf">is_cuda_backend</span><span class="p">(</span><span class="n">obj</span><span class="o">=</span><span class="kc">None</span><span class="p">):</span> |
| 780 | +<span class="w"> </span><span class="sd">"""</span> |
| 781 | +<span class="sd"> Checks that object has a CUDA backend.</span> |
| 782 | + |
| 783 | +<span class="sd"> Parameters</span> |
| 784 | +<span class="sd"> ----------</span> |
| 785 | +<span class="sd"> obj : {Device, SyclDevice, SyclQueue, dpnp.ndarray, usm_ndarray, None},</span> |
| 786 | +<span class="sd"> optional</span> |
| 787 | +<span class="sd"> An input object with sycl_device property to check device backend.</span> |
| 788 | +<span class="sd"> If `obj` is ``None``, device backend will be checked for the default</span> |
| 789 | +<span class="sd"> queue.</span> |
| 790 | +<span class="sd"> Default: ``None``.</span> |
| 791 | + |
| 792 | +<span class="sd"> Returns</span> |
| 793 | +<span class="sd"> -------</span> |
| 794 | +<span class="sd"> out : bool</span> |
| 795 | +<span class="sd"> Return ``True`` if data of the input object resides on a CUDA backend,</span> |
| 796 | +<span class="sd"> otherwise ``False``.</span> |
| 797 | + |
| 798 | +<span class="sd"> """</span> |
| 799 | + |
| 800 | + <span class="k">if</span> <span class="n">obj</span> <span class="ow">is</span> <span class="kc">None</span><span class="p">:</span> |
| 801 | + <span class="n">sycl_device</span> <span class="o">=</span> <span class="n">dpctl</span><span class="o">.</span><span class="n">select_default_device</span><span class="p">()</span> |
| 802 | + <span class="k">elif</span> <span class="nb">isinstance</span><span class="p">(</span><span class="n">obj</span><span class="p">,</span> <span class="n">dpctl</span><span class="o">.</span><span class="n">SyclDevice</span><span class="p">):</span> |
| 803 | + <span class="n">sycl_device</span> <span class="o">=</span> <span class="n">obj</span> |
| 804 | + <span class="k">else</span><span class="p">:</span> |
| 805 | + <span class="n">sycl_device</span> <span class="o">=</span> <span class="nb">getattr</span><span class="p">(</span><span class="n">obj</span><span class="p">,</span> <span class="s2">"sycl_device"</span><span class="p">,</span> <span class="kc">None</span><span class="p">)</span> |
| 806 | + <span class="k">if</span> <span class="p">(</span> |
| 807 | + <span class="n">sycl_device</span> <span class="ow">is</span> <span class="ow">not</span> <span class="kc">None</span> |
| 808 | + <span class="ow">and</span> <span class="n">sycl_device</span><span class="o">.</span><span class="n">backend</span> <span class="o">==</span> <span class="n">dpctl</span><span class="o">.</span><span class="n">backend_type</span><span class="o">.</span><span class="n">cuda</span> |
| 809 | + <span class="p">):</span> |
| 810 | + <span class="k">return</span> <span class="kc">True</span> |
| 811 | + <span class="k">return</span> <span class="kc">False</span> |
| 812 | + |
| 813 | + |
778 | 814 | <span class="k">def</span><span class="w"> </span><span class="nf">is_supported_array_or_scalar</span><span class="p">(</span><span class="n">a</span><span class="p">):</span>
|
779 | 815 | <span class="w"> </span><span class="sd">"""</span>
|
780 | 816 | <span class="sd"> Return ``True`` if `a` is a scalar or an array of either</span>
|
|
0 commit comments