Skip to content

Commit 4ebb1f5

Browse files
authored
fix an error in test_sparse_inner_product.cpp (#229)
1 parent b6b3d2d commit 4ebb1f5

File tree

1 file changed

+4
-3
lines changed

1 file changed

+4
-3
lines changed

nlp_toolkit/backends/neural_engine/test/gtest/test_sparse_inner_product.cpp

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -149,11 +149,12 @@ Tensor* make_int32_bias_obj(const TensorConfig* bias_tensor_config, const float*
149149
}
150150

151151
Tensor* get_fp32_dst(const TensorConfig* dst_tensor_config, vector<Tensor*> inputs) {
152-
// using namespace dnnl;
152+
using dnnl::matmul;
153+
using dnnl::memory;
153154
Tensor* dst_tensor = new Tensor(*dst_tensor_config);
154155
dst_tensor->add_tensor_life(1);
155-
engine engine(engine::kind::cpu, 0);
156-
stream engine_stream = stream(engine);
156+
dnnl::engine engine(dnnl::engine::kind::cpu, 0);
157+
dnnl::stream engine_stream = dnnl::stream(engine);
157158
Tensor* src = inputs[0];
158159
Tensor* weight = inputs[1];
159160
Tensor* bias = inputs[2];

0 commit comments

Comments
 (0)