Skip to content

Commit 710e17f

Browse files
committed
Fix get_mem_step_size function signature
1 parent 983500e commit 710e17f

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

arrayfire_wrapper/lib/create_and_modify_array/manage_device.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -143,11 +143,13 @@ def get_kernel_cache_directory(length: int, path: str, /) -> None:
143143
call_from_clib(get_kernel_cache_directory.__name__, ctypes.c_size_t(length), path.encode("utf-8"))
144144

145145

146-
def get_mem_step_size(step_bytes: int, /) -> None:
146+
def get_mem_step_size() -> int:
147147
"""
148148
source: https://arrayfire.org/docs/group__device__func__mem.htm#ga4c04df1ae248a6a8aa0a28263323872a
149149
"""
150-
call_from_clib(get_mem_step_size.__name__, ctypes.c_size_t(step_bytes))
150+
out = ctypes.c_int(0)
151+
call_from_clib(get_mem_step_size.__name__, ctypes.pointer(out))
152+
return int(out.value)
151153

152154

153155
def is_locked_array(arr: AFArray, /) -> bool:

0 commit comments

Comments
 (0)