Skip to content

Commit 074ab3d

Browse files
Handle allocation error for USM-host for strides
1 parent 859dc82 commit 074ab3d

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

dpctl/tensor/libtensor/source/clip.cpp

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -225,6 +225,9 @@ py_clip(const dpctl::tensor::usm_ndarray &src,
225225
simplified_shape, simplified_src_strides, simplified_min_strides,
226226
simplified_max_strides, simplified_dst_strides);
227227
py::ssize_t *packed_shape_strides = std::get<0>(ptr_size_event_tuple);
228+
if (!packed_shape_strides) {
229+
throw std::runtime_error("USM-host memory allocation failure");
230+
}
228231
sycl::event copy_shape_strides_ev = std::get<2>(ptr_size_event_tuple);
229232

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

0 commit comments

Comments
 (0)