Method test_raw

tools/Polygraphy/tests/cuda/test_cuda.py:175–181

(self)

        assert copy_to_time <= (memcpy_time * 1.05)

    def test_raw(self):
        with DeviceArray.raw((25,)) as buf:
            assert buf.shape == (25,)
            assert buf.nbytes == 25
            buf.resize((30,))
            assert buf.shape == (30,)
            assert buf.nbytes == 30