Skip to content

Commit b030a79

Browse files
committed
Handle allocation error for strides for where
1 parent 074ab3d commit b030a79

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

dpctl/tensor/libtensor/source/where.cpp

+3
Original file line numberDiff line numberDiff line change
@@ -203,6 +203,9 @@ py_where(const dpctl::tensor::usm_ndarray &condition,
203203
simplified_shape, simplified_cond_strides, simplified_x1_strides,
204204
simplified_x2_strides, simplified_dst_strides);
205205
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+
}
206209
sycl::event copy_shape_strides_ev = std::get<2>(ptr_size_event_tuple);
207210

208211
std::vector<sycl::event> all_deps;

0 commit comments

Comments
 (0)