12
12
</ script >
13
13
14
14
< meta name ="viewport " content ="width=device-width, initial-scale=1.0 " />
15
- < title > dpnp.dpnp_iface — Data Parallel Extension for NumPy 0.13.1dev3+12.g0e192159a2 documentation</ title >
15
+ < title > dpnp.dpnp_iface — Data Parallel Extension for NumPy 0.13.1dev3+29.g9c5184b9df documentation</ title >
16
16
< link rel ="stylesheet " type ="text/css " href ="../../_static/pygments.css?v=fa44fd50 " />
17
17
< link rel ="stylesheet " type ="text/css " href ="../../_static/css/theme.css?v=19f00094 " />
18
18
23
23
24
24
< script src ="../../_static/jquery.js?v=5d32c60e "> </ script >
25
25
< script src ="../../_static/_sphinx_javascript_frameworks_compat.js?v=2cd50e6c "> </ script >
26
- < script src ="../../_static/documentation_options.js?v=ce66c2d4 "> </ script >
26
+ < script src ="../../_static/documentation_options.js?v=31b1a6b4 "> </ script >
27
27
< script src ="../../_static/doctools.js?v=888ff710 "> </ script >
28
28
< script src ="../../_static/sphinx_highlight.js?v=dc90522c "> </ script >
29
29
< script src ="../../_static/js/theme.js "> </ script >
43
43
Data Parallel Extension for NumPy
44
44
</ a >
45
45
< div class ="version ">
46
- 0.13.1dev3+12.g0e192159a2
46
+ 0.13.1dev3+29.g9c5184b9df
47
47
</ div >
48
48
< div role ="search ">
49
49
< form id ="rtd-search-form " class ="wy-form " action ="../../search.html " method ="get ">
@@ -301,7 +301,7 @@ <h1>Source code for dpnp.dpnp_iface</h1><div class="highlight"><pre>
301
301
< 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 "> array_obj</ span > < span class ="p "> )</ span >
302
302
303
303
304
- < span class ="k "> def</ span > < span class ="nf "> check_supported_arrays_type</ span > < span class ="p "> (</ span > < span class ="o "> *</ span > < span class ="n "> arrays</ span > < span class ="p "> ,</ span > < span class ="n "> scalar_type</ span > < span class ="o "> =</ span > < span class ="kc "> False</ span > < span class ="p "> ):</ span >
304
+ < span class ="k "> def</ span > < span class ="nf "> check_supported_arrays_type</ span > < span class ="p "> (</ span > < span class ="o "> *</ span > < span class ="n "> arrays</ span > < span class ="p "> ,</ span > < span class ="n "> scalar_type</ span > < span class ="o "> =</ span > < span class ="kc "> False</ span > < span class ="p "> , </ span > < span class =" n " > all_scalars </ span > < span class =" o " > = </ span > < span class =" kc " > False </ span > < span class =" p " > ):</ span >
305
305
< span class ="w "> </ span > < span class ="sd "> """</ span >
306
306
< span class ="sd "> Return ``True`` if each array has either type of scalar,</ span >
307
307
< span class ="sd "> :class:`dpnp.ndarray` or :class:`dpctl.tensor.usm_ndarray`.</ span >
@@ -312,7 +312,9 @@ <h1>Source code for dpnp.dpnp_iface</h1><div class="highlight"><pre>
312
312
< span class ="sd "> arrays : {dpnp_array, usm_ndarray}</ span >
313
313
< span class ="sd "> Input arrays to check for supported types.</ span >
314
314
< span class ="sd "> scalar_type : {bool}, optional</ span >
315
- < span class ="sd "> A scalar type is also considered as supported if flag is True.</ span >
315
+ < span class ="sd "> A scalar type is also considered as supported if flag is ``True``.</ span >
316
+ < span class ="sd "> all_scalars : {bool}, optional</ span >
317
+ < span class ="sd "> All the input arrays can be scalar if flag is ``True``.</ span >
316
318
317
319
< span class ="sd "> Returns</ span >
318
320
< span class ="sd "> -------</ span >
@@ -327,13 +329,22 @@ <h1>Source code for dpnp.dpnp_iface</h1><div class="highlight"><pre>
327
329
328
330
< span class ="sd "> """</ span >
329
331
332
+ < span class ="n "> any_is_array</ span > < span class ="o "> =</ span > < span class ="kc "> False</ span >
330
333
< span class ="k "> for</ span > < span class ="n "> a</ span > < span class ="ow "> in</ span > < span class ="n "> arrays</ span > < span class ="p "> :</ span >
331
- < span class ="k "> if</ span > < span class ="n "> scalar_type</ span > < span class ="ow "> and</ 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 ="ow "> or</ span > < span class ="n "> is_supported_array_type</ span > < span class ="p "> (</ span > < span class ="n "> a</ span > < span class ="p "> ):</ span >
334
+ < span class ="k "> if</ span > < span class ="n "> is_supported_array_type</ span > < span class ="p "> (</ span > < span class ="n "> a</ span > < span class ="p "> ):</ span >
335
+ < span class ="n "> any_is_array</ span > < span class ="o "> =</ span > < span class ="kc "> True</ span >
336
+ < span class ="k "> continue</ span >
337
+ < span class ="k "> elif</ span > < span class ="n "> scalar_type</ span > < span class ="ow "> and</ 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 >
332
338
< span class ="k "> continue</ span >
333
339
334
340
< span class ="k "> raise</ span > < span class ="ne "> TypeError</ span > < span class ="p "> (</ span >
335
341
< span class ="s2 "> "An array must be any of supported type, but got </ span > < span class ="si "> {}</ span > < span class ="s2 "> "</ span > < span class ="o "> .</ span > < span class ="n "> format</ span > < span class ="p "> (</ span > < span class ="nb "> type</ span > < span class ="p "> (</ span > < span class ="n "> a</ span > < span class ="p "> ))</ span >
336
342
< span class ="p "> )</ span >
343
+
344
+ < span class ="k "> if</ span > < span class ="nb "> len</ span > < span class ="p "> (</ span > < span class ="n "> arrays</ span > < span class ="p "> )</ span > < span class ="o "> ></ span > < span class ="mi "> 1</ span > < span class ="ow "> and</ span > < span class ="ow "> not</ span > < span class ="p "> (</ span > < span class ="n "> all_scalars</ span > < span class ="ow "> or</ span > < span class ="n "> any_is_array</ span > < span class ="p "> ):</ span >
345
+ < span class ="k "> raise</ span > < span class ="ne "> TypeError</ span > < span class ="p "> (</ span >
346
+ < span class ="s2 "> "At least one input must be of supported array type, but got all scalars."</ span >
347
+ < span class ="p "> )</ span >
337
348
< span class ="k "> return</ span > < span class ="kc "> True</ span >
338
349
339
350
@@ -586,17 +597,20 @@ <h1>Source code for dpnp.dpnp_iface</h1><div class="highlight"><pre>
586
597
< span class ="k "> if</ span > < span class ="n "> out</ span > < span class ="ow "> is</ span > < span class ="kc "> None</ span > < span class ="p "> :</ span >
587
598
< span class ="k "> return</ span > < span class ="n "> a</ span >
588
599
< span class ="k "> else</ span > < span class ="p "> :</ span >
589
- < span class ="n "> dpnp</ span > < span class ="o "> .</ span > < span class ="n "> check_supported_arrays_type</ span > < span class ="p "> (</ span > < span class ="n "> out</ span > < span class ="p "> )</ span >
590
- < span class ="k "> if</ span > < span class ="n "> out</ span > < span class ="o "> .</ span > < span class ="n "> shape</ span > < span class ="o "> !=</ span > < span class ="n "> a</ span > < span class ="o "> .</ span > < span class ="n "> shape</ span > < span class ="p "> :</ span >
591
- < span class ="k "> raise</ span > < span class ="ne "> ValueError</ span > < span class ="p "> (</ span >
592
- < span class ="sa "> f</ span > < span class ="s2 "> "Output array of shape </ span > < span class ="si "> {</ span > < span class ="n "> a</ span > < span class ="o "> .</ span > < span class ="n "> shape</ span > < span class ="si "> }</ span > < span class ="s2 "> is needed, got </ span > < span class ="si "> {</ span > < span class ="n "> out</ span > < span class ="o "> .</ span > < span class ="n "> shape</ span > < span class ="si "> }</ span > < span class ="s2 "> ."</ span >
593
- < span class ="p "> )</ span >
594
- < span class ="k "> elif</ span > < span class ="nb "> isinstance</ span > < span class ="p "> (</ span > < span class ="n "> out</ span > < span class ="p "> ,</ span > < span class ="n "> dpt</ span > < span class ="o "> .</ span > < span class ="n "> usm_ndarray</ span > < span class ="p "> ):</ span >
595
- < span class ="n "> out</ span > < span class ="o "> =</ 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 "> out</ span > < span class ="p "> )</ span >
600
+ < span class ="k "> if</ span > < span class ="n "> a</ span > < span class ="ow "> is</ span > < span class ="n "> out</ span > < span class ="p "> :</ span >
601
+ < span class ="k "> return</ span > < span class ="n "> out</ span >
602
+ < span class ="k "> else</ span > < span class ="p "> :</ span >
603
+ < span class ="n "> dpnp</ span > < span class ="o "> .</ span > < span class ="n "> check_supported_arrays_type</ span > < span class ="p "> (</ span > < span class ="n "> out</ span > < span class ="p "> )</ span >
604
+ < span class ="k "> if</ span > < span class ="n "> out</ span > < span class ="o "> .</ span > < span class ="n "> shape</ span > < span class ="o "> !=</ span > < span class ="n "> a</ span > < span class ="o "> .</ span > < span class ="n "> shape</ span > < span class ="p "> :</ span >
605
+ < span class ="k "> raise</ span > < span class ="ne "> ValueError</ span > < span class ="p "> (</ span >
606
+ < span class ="sa "> f</ span > < span class ="s2 "> "Output array of shape </ span > < span class ="si "> {</ span > < span class ="n "> a</ span > < span class ="o "> .</ span > < span class ="n "> shape</ span > < span class ="si "> }</ span > < span class ="s2 "> is needed, got </ span > < span class ="si "> {</ span > < span class ="n "> out</ span > < span class ="o "> .</ span > < span class ="n "> shape</ span > < span class ="si "> }</ span > < span class ="s2 "> ."</ span >
607
+ < span class ="p "> )</ span >
608
+ < span class ="k "> elif</ span > < span class ="nb "> isinstance</ span > < span class ="p "> (</ span > < span class ="n "> out</ span > < span class ="p "> ,</ span > < span class ="n "> dpt</ span > < span class ="o "> .</ span > < span class ="n "> usm_ndarray</ span > < span class ="p "> ):</ span >
609
+ < span class ="n "> out</ span > < span class ="o "> =</ 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 "> out</ span > < span class ="p "> )</ span >
596
610
597
- < span class ="n "> dpnp</ span > < span class ="o "> .</ span > < span class ="n "> copyto</ span > < span class ="p "> (</ span > < span class ="n "> out</ span > < span class ="p "> ,</ span > < span class ="n "> a</ span > < span class ="p "> ,</ span > < span class ="n "> casting</ span > < span class ="o "> =</ span > < span class ="n "> casting</ span > < span class ="p "> )</ span >
611
+ < span class ="n "> dpnp</ span > < span class ="o "> .</ span > < span class ="n "> copyto</ span > < span class ="p "> (</ span > < span class ="n "> out</ span > < span class ="p "> ,</ span > < span class ="n "> a</ span > < span class ="p "> ,</ span > < span class ="n "> casting</ span > < span class ="o "> =</ span > < span class ="n "> casting</ span > < span class ="p "> )</ span >
598
612
599
- < span class ="k "> return</ span > < span class ="n "> out</ span >
613
+ < span class ="k "> return</ span > < span class ="n "> out</ span >
600
614
601
615
602
616
< span class ="k "> def</ span > < span class ="nf "> get_usm_ndarray</ span > < span class ="p "> (</ span > < span class ="n "> a</ span > < span class ="p "> ):</ span >
0 commit comments