p35189024
/
InfiniTensor
forked from jiuyuan/InfiniTensor
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

fix: test graph handler 

Signed-off-by: YdrMaster <ydrml@hotmail.com>
This commit is contained in:
YdrMaster 2023-04-20 16:05:33 +08:00 committed by Liyan Zheng
parent 28b123753e
commit 8bc2d3e48d
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
test/core/test_graph_handler.cc
View File

@ -8,8 +8,9 @@ TEST(Handler, matmul) {
auto runtime = NativeCpuRuntimeObj::getInstance();
auto handler = make_ref<GraphHandlerObj>(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);
}