saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-06-04 23:59:56 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Fix x86 build break (#5759)

Browse source
This commit is contained in:
Guoyu Wang 2020-11-10 20:33:27 -08:00 committed by GitHub
parent 042365029f
commit 0767c4fdfb
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
onnxruntime/core/providers/cpu/nn/qlinearconv.cc
View file

@ -88,10 +88,6 @@ Status QLinearConv::PrePack(const Tensor& tensor, int input_idx, bool& is_packed
const size_t kernel_size =
static_cast<size_t>(std::accumulate(shape.data() + 2, shape.data() + rank, 1LL, std::multiplies<int64_t>()));
const size_t group_count = static_cast<size_t>(conv_attrs_.group);
const size_t group_output_channels = output_channels / group_count;
const size_t kernel_dim = group_input_channels * kernel_size;
const auto* Wdata = static_cast<const uint8_t*>(tensor.DataRaw());
W_shape_ = shape;
is_W_signed_ = tensor.IsDataType<int8_t>();
@ -99,6 +95,10 @@ Status QLinearConv::PrePack(const Tensor& tensor, int input_idx, bool& is_packed
auto alloc = Info().GetAllocator(0, OrtMemTypeDefault);
#ifdef MLAS_SUPPORTS_PACKED_GEMM_U8X8
const size_t group_count = static_cast<size_t>(conv_attrs_.group);
const size_t group_output_channels = output_channels / group_count;
const size_t kernel_dim = group_input_channels * kernel_size;
// Don't pack the filter buffer if the MlasConvDepthwise path is used.
if (group_input_channels != 1 && group_output_channels != 1) {
packed_W_size_ = MlasGemmPackBSize(group_output_channels, kernel_dim, true);