def test_broadcast_to(): shape = (200, 300) to_shape = (130, 200, 300) x = np.random.uniform(-1, 1, shape).astype(dtype) y = np.zeros(to_shape).astype(dtype) arr_x = tvm.nd.array(x, ctx=ctx) arr_y = tvm.nd.array(y, ctx=ctx) broadcast_to = tvm_op.make_broadcast_to(shape, to_shape, tgt, tgt_host, "broadcast_to") broadcast_to(arr_x, arr_y) y = arr_y.asnumpy() np.testing.assert_allclose(np.broadcast_to(x, to_shape), y)