Skip to content

Commit 4b3369f

Browse files
committed
Adding get_size_of
1 parent ef44a7a commit 4b3369f

File tree

1 file changed

+8
-0
lines changed

1 file changed

+8
-0
lines changed

arrayfire/library.py

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -671,4 +671,12 @@ def get_device_id(A):
671671
safe_call(backend.get().af_get_device_id(ct.pointer(device_id), A.arr))
672672
return device_id
673673

674+
def get_size_of(dtype):
675+
"""
676+
Get the size of the type represented by arrayfire.Dtype
677+
"""
678+
size = ct.c_size_t(0)
679+
safe_call(backend.get().af_get_size_of(ct.pointer(size), dtype.value))
680+
return size.value
681+
674682
from .util import safe_call

0 commit comments

Comments
 (0)