Skip to content

GitLab

Commit 3a6d8366 authored by zhangyue's avatar zhangyue
Browse files

issue/87: fix return in matmul calculate

parent 1df7701e
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 2 deletions
src/infiniop/ops/matmul/kunlun/matmul_kunlun.cc
View file @ 3a6d8366
...@@ -61,7 +61,7 @@ infiniStatus_t calculate( ...@@ -61,7 +61,7 @@ infiniStatus_t calculate(
auto unit = infiniSizeOf(dtype); auto unit = infiniSizeOf(dtype);
return internal->useXdnn( CHECK_STATUS(internal->useXdnn(
(kunlunStream_t)stream, (kunlunStream_t)stream,
[&](xdnnHandle_t handle) { [&](xdnnHandle_t handle) {
for (size_t i = 0; i < info.batch; i++) { for (size_t i = 0; i < info.batch; i++) {
...@@ -88,7 +88,8 @@ infiniStatus_t calculate( ...@@ -88,7 +88,8 @@ infiniStatus_t calculate(
nullptr))); nullptr)));
} }
return INFINI_STATUS_SUCCESS; return INFINI_STATUS_SUCCESS;
}); }));
return INFINI_STATUS_SUCCESS;
} }
infiniStatus_t Descriptor::calculate( infiniStatus_t Descriptor::calculate(
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment