Создание массива Cupy из существующего указателя графического процессора ⇐ Python

[Anonymous]

Я хотел бы создать представление массива графического процессора Cupy для массива, который уже существует в графическом процессоре, и мне было передано следующее:

• Указатель на массив.
• Я знаю тип данных и размер данных.
• Мне также дана презентация.

Как можно построить представление массива (желательно избегать копий)? Я попробовал следующее:

Код: Выделить всё

import cupy as cp
import numpy as np

shape = (w, h, c, b) # example
s = np.product(shape)*4 # this is 1D

mem = cp.cuda.UnownedMemory(ptr=image_batch_ptr,
owner=None,
size=s)
memptr = cp.cuda.MemoryPointer(mem, 0)
d = cp.ndarray(shape=shape,
dtype=np.float32,
memptr=memptr)

Но, похоже, это не приводит к правильному выравниванию. В частности, у меня проблемы с интеграцией высоты тона в изображение. Возможно ли это вообще?


Подробнее здесь: https://stackoverflow.com/questions/713 ... pu-pointer