Function py_buffer_from_address

dali/test/python/test_utils.py:581–596

(address, shape, dtype, gpu=False)

def py_buffer_from_address(address, shape, dtype, gpu=False):
    import_numpy()

    buff = {"data": (address, False), "shape": tuple(shape), "typestr": np.dtype(dtype).str}

    class py_holder(object):
        pass

    holder = py_holder()
    holder.__array_interface__ = buff
    holder.__cuda_array_interface__ = buff
    if not gpu:
        return np.array(holder, copy=False)
    else:
        import_cupy()
        return cp.asanyarray(holder)


class check_output_pattern: