Browse Source

!30213 [MSLITE][DEVELOP] fix VS bug

Merge pull request !30213 from yangruoqi713/master
feature/build-system-rewrite
i-robot Gitee 4 years ago
parent
commit
e4a86db2e8
No known key found for this signature in database GPG Key ID: 173E9B9CA92EEF8F
4 changed files with 7 additions and 6 deletions
  1. +3
    -3
      mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/fp32/winograd_utils.c
  2. +2
    -1
      mindspore/core/mindrt/src/thread/core_affinity.cc
  3. +1
    -1
      mindspore/lite/src/ops/populate/custom_populate.cc
  4. +1
    -1
      mindspore/lite/src/runtime/kernel/arm/base/quant_dtype_cast.cc

+ 3
- 3
mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/fp32/winograd_utils.c View File

@@ -262,7 +262,7 @@ void InputTransform4x4Pack12(float *src_data, float *dst_data, int src_step, int
for (int l = 0; l < 4; ++l) {
float *src = src_data + l * pack_tile * block_tile;
// 12 * 4 -> 4 * 12
float tmp_mat[pack_tile][block_tile];
float tmp_mat[4][12];
for (int i = 0; i < block_tile; ++i) {
for (int j = 0; j < pack_tile; ++j) {
tmp_mat[j][i] = src[i * pack_tile + j];
@@ -500,7 +500,7 @@ void InputTransform6x6Pack12(float *src_data, float *dst_data, int src_step, int
for (int l = 0; l < 6; ++l) {
float *src = src_data + l * pack_tile * block_tile;
// 12 * 4 -> 4 * 12
float tmp_mat[pack_tile][block_tile];
float tmp_mat[4][12];
for (int i = 0; i < block_tile; ++i) {
for (int j = 0; j < pack_tile; ++j) {
tmp_mat[j][i] = src[i * pack_tile + j];
@@ -838,7 +838,7 @@ void InputTransform8x8Pack12(float *src_data, float *dst_data, int src_step, int
for (int l = 0; l < 8; ++l) {
float *src = src_data + l * pack_tile * block_tile;
// 12 * 4 -> 4 * 12
float tmp_mat[pack_tile][block_tile];
float tmp_mat[4][12];
for (int i = 0; i < block_tile; ++i) {
for (int j = 0; j < pack_tile; ++j) {
tmp_mat[j][i] = src[i * pack_tile + j];


+ 2
- 1
mindspore/core/mindrt/src/thread/core_affinity.cc View File

@@ -16,7 +16,8 @@

#include "thread/core_affinity.h"
#include <string.h>
#include <stdlib.h>
#include <cstdlib>
#include <string>
#include <algorithm>
#ifdef MS_COMPILE_IOS
#include <sys/types.h>


+ 1
- 1
mindspore/lite/src/ops/populate/custom_populate.cc View File

@@ -16,7 +16,7 @@
#include "src/ops/populate/populate_register.h"
#include "src/common/log_adapter.h"
#include "src/tensor.h"
using PrimType::PrimType_Custom;
using mindspore::schema::PrimitiveType_Custom;

namespace mindspore {
namespace lite {


+ 1
- 1
mindspore/lite/src/runtime/kernel/arm/base/quant_dtype_cast.cc View File

@@ -107,7 +107,7 @@ int QuantDTypeCastCPUKernel::QuantDTypeCast(int task_id) {
auto input_quant_arg = in_tensors_.front()->quant_params().front();
ret = DoDequantizeInt8ToFp32(int8_ptr_ + thread_offset, float32_ptr_ + thread_offset, input_quant_arg.scale,
input_quant_arg.zeroPoint, num_unit_thread);
if (ret) {
if (ret == RET_OK) {
auto output_quant_arg = out_tensors_.front()->quant_params().front();
if (quant_arg.dstDtype == TypeId::kNumberTypeUInt8) {
ret = DoQuantizeFp32ToInt8FromUint8Source(float32_ptr_ + thread_offset, int8_out_ptr_ + thread_offset,


Loading…
Cancel
Save