We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 074ab3d commit b030a79Copy full SHA for b030a79
dpctl/tensor/libtensor/source/where.cpp
@@ -203,6 +203,9 @@ py_where(const dpctl::tensor::usm_ndarray &condition,
203
simplified_shape, simplified_cond_strides, simplified_x1_strides,
204
simplified_x2_strides, simplified_dst_strides);
205
py::ssize_t *packed_shape_strides = std::get<0>(ptr_size_event_tuple);
206
+ if (!packed_shape_strides) {
207
+ throw std::runtime_error("USM-host memory allocation failure");
208
+ }
209
sycl::event copy_shape_strides_ev = std::get<2>(ptr_size_event_tuple);
210
211
std::vector<sycl::event> all_deps;
0 commit comments