本文整理匯總了Python中cuda4py._py.CU.extract_ptr_and_size方法的典型用法代碼示例。如果您正苦於以下問題：Python CU.extract_ptr_and_size方法的具體用法？Python CU.extract_ptr_and_size怎麽用？Python CU.extract_ptr_and_size使用的例子？那麽, 這裏精選的方法代碼示例或許可以為您提供幫助。您也可以進一步了解該方法所在類cuda4py._py.CU的用法示例。
在下文中一共展示了CU.extract_ptr_and_size方法的1個代碼示例，這些例子默認根據受歡迎程度排序。您可以為喜歡或者感覺有用的代碼點讚，您的評價將有助於係統推薦出更棒的Python代碼示例。
示例1: _extract_ptr_and_count
# 需要導入模塊: from cuda4py._py import CU [as 別名]
# 或者: from cuda4py._py.CU import extract_ptr_and_size [as 別名]
def _extract_ptr_and_count(self, arr, count, itemsize):
"""Returns tuple of address of an arr and extracted item count
casted to int.
It will clamp requested count to an array size if possible.
"""
if self.context is None:
arr, size = CU.extract_ptr_and_size(arr, None)
elif count is None:
size = arr.size
else:
size = getattr(arr, "size", count * itemsize)
size = size if count is None else min(count * itemsize, size)
return int(arr), int(size) // itemsize