From 8bc2d3e48deb280f9b3cc49c717423d337e99c1e Mon Sep 17 00:00:00 2001 From: YdrMaster Date: Thu, 20 Apr 2023 16:05:33 +0800 Subject: [PATCH] fix: test graph handler Signed-off-by: YdrMaster --- test/core/test_graph_handler.cc | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/test/core/test_graph_handler.cc b/test/core/test_graph_handler.cc index 8c25568c..b19c9479 100644 --- a/test/core/test_graph_handler.cc +++ b/test/core/test_graph_handler.cc @@ -8,8 +8,9 @@ TEST(Handler, matmul) { auto runtime = NativeCpuRuntimeObj::getInstance(); auto handler = make_ref(runtime); auto i = handler->tensor({1, 2, 3}, OnnxDType::UINT32, TensorType::Input); - auto w = handler->tensor({1, 3, 4}, OnnxDType::UINT32, TensorType::Input); - auto o = handler->tensor({1, 2, 4}, OnnxDType::UINT32, TensorType::Input); + auto w = + handler->tensor({1, 3, 4}, OnnxDType::UINT32, TensorType::Initialized); + auto o = handler->tensor({1, 2, 4}, OnnxDType::UINT32, TensorType::Other); handler->matmul(i, w, o, false, false, nullptr, ActType::None); }