Skip to content

Commit

Permalink
fix build error
Browse files Browse the repository at this point in the history
  • Loading branch information
hans00 committed May 30, 2024
1 parent e224328 commit b0a7542
Showing 1 changed file with 4 additions and 0 deletions.
4 changes: 4 additions & 0 deletions onnxruntime/core/providers/xnnpack/math/matmul_int.cc
Original file line number Diff line number Diff line change
Expand Up @@ -70,6 +70,7 @@ bool MatMulIntegerCommon::IsOnnxNodeSupported(const NodeUnit& node_unit, const G
return supported;
}

template<>
Status MatMulInteger<int8_t>::PrePack(const Tensor& tensor, int input_idx, AllocatorPtr alloc,
/*out*/ bool& is_packed,
/*out*/ PrePackedWeights* /*Not used*/) {
Expand Down Expand Up @@ -136,6 +137,7 @@ Status MatMulInteger<int8_t>::PrePack(const Tensor& tensor, int input_idx, Alloc
return Status::OK();
}

template<>
Status MatMulInteger<uint8_t>::PrePack(const Tensor& tensor, int input_idx, AllocatorPtr alloc,
/*out*/ bool& is_packed,
/*out*/ PrePackedWeights* /*Not used*/) {
Expand Down Expand Up @@ -204,6 +206,7 @@ Status MatMulInteger<uint8_t>::PrePack(const Tensor& tensor, int input_idx, Allo
return Status::OK();
}

template<>
Status MatMulInteger<int8_t>::Compute(OpKernelContext* ctx) const {
const Tensor* a = ctx->Input<Tensor>(0);
pthreadpool_t threadpool = GetThreadPool();
Expand Down Expand Up @@ -249,6 +252,7 @@ Status MatMulInteger<int8_t>::Compute(OpKernelContext* ctx) const {
return Status::OK();
}

template<>
Status MatMulInteger<uint8_t>::Compute(OpKernelContext* ctx) const {
const Tensor* a = ctx->Input<Tensor>(0);
pthreadpool_t threadpool = GetThreadPool();
Expand Down

0 comments on commit b0a7542

Please sign in to comment.