@@ -904,8 +904,6 @@ <h1>Source code for dpctl.tensor._ctors</h1><div class="highlight"><pre>
904904 < span class ="k "> raise</ span > < span class ="ne "> TypeError</ span > < span class ="p "> (</ span >
905905 < span class ="sa "> f</ span > < span class ="s2 "> "Expected dpctl.tensor.usm_ndarray, got </ span > < span class ="si "> {</ span > < span class ="nb "> type</ span > < span class ="p "> (</ span > < span class ="n "> usm_ndary</ span > < span class ="p "> )</ span > < span class ="si "> }</ span > < span class ="s2 "> "</ span >
906906 < span class ="p "> )</ span >
907- < span class ="k "> if</ span > < span class ="n "> dtype</ span > < span class ="ow "> is</ span > < span class ="kc "> None</ span > < span class ="p "> :</ span >
908- < span class ="n "> dtype</ span > < span class ="o "> =</ span > < span class ="n "> usm_ndary</ span > < span class ="o "> .</ span > < span class ="n "> dtype</ span >
909907 < span class ="k "> if</ span > < span class ="n "> usm_type</ span > < span class ="ow "> is</ span > < span class ="kc "> None</ span > < span class ="p "> :</ span >
910908 < span class ="n "> usm_type</ span > < span class ="o "> =</ span > < span class ="n "> usm_ndary</ span > < span class ="o "> .</ span > < span class ="n "> usm_type</ span >
911909 < span class ="k "> if</ span > < span class ="n "> sycl_queue</ span > < span class ="ow "> is</ span > < span class ="ow "> not</ span > < span class ="kc "> None</ span > < span class ="p "> :</ span >
@@ -915,6 +913,8 @@ <h1>Source code for dpctl.tensor._ctors</h1><div class="highlight"><pre>
915913 < span class ="n "> copy_q</ span > < span class ="o "> =</ span > < span class ="n "> normalize_queue_device</ span > < span class ="p "> (</ span > < span class ="n "> sycl_queue</ span > < span class ="o "> =</ span > < span class ="n "> sycl_queue</ span > < span class ="p "> ,</ span > < span class ="n "> device</ span > < span class ="o "> =</ span > < span class ="n "> exec_q</ span > < span class ="p "> )</ span >
916914 < span class ="k "> else</ span > < span class ="p "> :</ span >
917915 < span class ="n "> copy_q</ span > < span class ="o "> =</ span > < span class ="n "> usm_ndary</ span > < span class ="o "> .</ span > < span class ="n "> sycl_queue</ span >
916+ < span class ="k "> if</ span > < span class ="n "> dtype</ span > < span class ="ow "> is</ span > < span class ="kc "> None</ span > < span class ="p "> :</ span >
917+ < span class ="n "> dtype</ span > < span class ="o "> =</ span > < span class ="n "> _map_to_device_dtype</ span > < span class ="p "> (</ span > < span class ="n "> usm_ndary</ span > < span class ="o "> .</ span > < span class ="n "> dtype</ span > < span class ="p "> ,</ span > < span class ="n "> copy_q</ span > < span class ="p "> )</ span >
918918 < span class ="c1 "> # Conditions for zero copy:</ span >
919919 < span class ="n "> can_zero_copy</ span > < span class ="o "> =</ span > < span class ="n "> copy</ span > < span class ="ow "> is</ span > < span class ="ow "> not</ span > < span class ="kc "> True</ span >
920920 < span class ="c1 "> # dtype is unchanged</ span >
0 commit comments