Skip to content

Commit 8aa3dc9

Browse files
Write _set_writable_flag using helper function _copy_writable
1 parent 356cf22 commit 8aa3dc9

File tree

1 file changed

+2
-4
lines changed

1 file changed

+2
-4
lines changed

dpctl/tensor/_usmarray.pyx

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -586,10 +586,8 @@ cdef class usm_ndarray:
586586
return _flags.Flags(self, self.flags_)
587587

588588
cdef _set_writable_flag(self, int flag):
589-
cdef int arr_fl = self.flags_
590-
arr_fl ^= (arr_fl & USM_ARRAY_WRITABLE) # unset WRITABLE flag
591-
arr_fl |= (USM_ARRAY_WRITABLE if flag else 0)
592-
self.flags_ = arr_fl
589+
cdef int mask = (USM_ARRAY_WRITABLE if flag else 0)
590+
self.flags_ = _copy_writable(self.flags_, mask)
593591

594592
@property
595593
def usm_type(self):

0 commit comments

Comments
 (0)