Method test_basic

test/cinn/test_hlir_framework.py:24–33

(self)

class TensorTest(unittest.TestCase):
    def test_basic(self):
        target = Target()
        target.arch = Target.X86Arch()
        target.bits = Target.Bit.k64
        target.os = Target.OS.Linux
        tensor = Tensor()
        data = np.random.random([10, 5])
        tensor.from_numpy(data, target)

        np.testing.assert_allclose(tensor.numpy(), data)


if __name__ == "__main__":