Skip to content

Commit 8691551

Browse files
Extend check_no_overlap to handle same logical tensors
1 parent 819262d commit 8691551

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

dpnp/backend/extensions/common/ext/details/validation_utils_internal.hpp

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -114,8 +114,10 @@ inline void check_no_overlap(const array_ptr &input,
114114
}
115115

116116
const auto &overlap = dpctl::tensor::overlap::MemoryOverlap();
117+
const auto &same_logical_tensors =
118+
dpctl::tensor::overlap::SameLogicalTensors();
117119

118-
if (overlap(*input, *output)) {
120+
if (overlap(*input, *output) && !same_logical_tensors(*input, *output)) {
119121
throw py::value_error(name_of(input, names) +
120122
" has overlapping memory segments with " +
121123
name_of(output, names));

0 commit comments

Comments
 (0)