Skip to content

Commit 72737f8

Browse files
committed
add info properties to MKLMemory
1 parent d4cc27f commit 72737f8

1 file changed

Lines changed: 24 additions & 0 deletions

File tree

mkl/_mkl_memory.pyx

Lines changed: 24 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -175,3 +175,27 @@ cdef class MKLMemory:
175175

176176
self._memory_ptr = p
177177
self.nbytes = new_nbytes
178+
179+
@property
180+
def nbytes(self):
181+
return self.nbytes
182+
183+
@property
184+
def size(self):
185+
return self.nbytes
186+
187+
@property
188+
def _pointer(self):
189+
return <size_t>(self._memory_ptr)
190+
191+
def __repr__(self):
192+
return (
193+
f"<MKL memory allocation of {self.nbytes} bytes at "
194+
f"{hex(<object>(<size_t>self._memory_ptr))}>"
195+
)
196+
197+
def __len__(self):
198+
return self.nbytes
199+
200+
def __sizeof__(self):
201+
return self.nbytes

0 commit comments

Comments
 (0)