Method test_basic

tools/Polygraphy/tests/cuda/test_cuda.py:26–32

(self)

class TestDeviceView:
    def test_basic(self):
        with DeviceArray(shape=(1, 4, 2), dtype=np.float32) as arr:
            v = DeviceView(arr.ptr, arr.shape, arr.dtype)
            assert v.ptr == arr.ptr
            assert v.shape == arr.shape
            assert v.dtype == arr.dtype
            assert v.nbytes == arr.nbytes

    def test_with_int_ptr(self):
        ptr = 74892