Shortcuts

hfai.cuda

pin_memory

使用 cuda 内置函数, 把符合 Buffer Protocol 的对象放入锁页内存

unpin_memory

使用 cuda 内置函数, 将已放入锁页内存的对象解除锁页

hfai.cuda.pin_memory(data)[source]

使用 cuda 内置函数, 把符合 Buffer Protocol 的对象放入锁页内存

Parameters

data (any) – 要放入锁页内存的对象

Returns

被放入锁页内存的对象

Return type

any

Examples:

from hfai.cuda import pin_memory
pinned_data = pin_memory(unpinned_data)
hfai.cuda.unpin_memory(data)[source]

使用 cuda 内置函数, 将已放入锁页内存的对象解除锁页

Parameters

data (any) – 需要被解除锁页的对象

Returns

被解除锁页的对象

Return type

any

Examples:

from hfai.cuda import unpin_memory
unpinned_data = unpin_memory(pinned_data)