saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-07-26 19:52:38 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Fix build issues

Browse source
This commit is contained in:
Jeff 2020-04-13 08:36:14 -07:00
parent b04e48333d
commit e3c9fb9fee
2 changed files with 3 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
onnxruntime/core/providers/cpu/reduction/reduction_ops.cc
View file

@ -566,7 +566,9 @@ Status ReduceSum<T>::Compute(OpKernelContext* ctx) const {
if (no_transpose) {
const T* input_data = ctx->Input<Tensor>(0)->template Data<T>();
auto lambda = [input_data, blocks, output_data](ptrdiff_t i) {
output_data[i] = ConstEigenVectorMap<T>(input_data + (i * blocks), blocks).sum();
// The ConstEigenMatrixMap type is expanded to work around a MS compiler issue
output_data[i] = Eigen::Map<const Eigen::Matrix<T, Eigen::Dynamic, 1>>(input_data + (i * blocks), blocks).sum();
};
concurrency::ThreadPool::TryBatchParallelFor(ctx->GetOperatorThreadPool(), block_size, lambda, 0);
} else {

3
winml/test/common/taefTestMacros.h
View file

@ -1,8 +1,5 @@
// Copyright (c) Microsoft Corporation. All rights reserved.
// Licensed under the MIT License.
#define INLINE_TEST_METHOD_MARKUP
#include "WexTestClass.h"
using namespace WEX::Logging;