Browse Source

!230 fix clang complier bug

Merge pull request !230 from SanjayChan/master
tags/v0.2.0-alpha
mindspore-ci-bot Gitee 5 years ago
parent
commit
5ec3c8a215
4 changed files with 5 additions and 4 deletions
  1. +2
    -2
      mindspore/ccsrc/onnx/onnx_exporter.cc
  2. +1
    -1
      mindspore/ccsrc/optimizer/irpass/incorporate_getitem.h
  3. +1
    -1
      mindspore/ccsrc/parallel/ops_info/operator_info.cc
  4. +1
    -0
      mindspore/ccsrc/utils/log_adapter.h

+ 2
- 2
mindspore/ccsrc/onnx/onnx_exporter.cc View File

@@ -174,14 +174,14 @@ OPERATOR_ONNX_CONVERT_DEFINE(Sigmoid, Sigmoid, OpNameInfo())
OPERATOR_ONNX_CONVERT_DEFINE(Flatten, Flatten, OpNameInfo()) OPERATOR_ONNX_CONVERT_DEFINE(Flatten, Flatten, OpNameInfo())
OPERATOR_ONNX_CONVERT_DEFINE(Squeeze, Squeeze, OPERATOR_ONNX_CONVERT_DEFINE(Squeeze, Squeeze,
OpNameInfo().Attr("axis", "axes", onnx::AttributeProto_AttributeType_INTS, OpNameInfo().Attr("axis", "axes", onnx::AttributeProto_AttributeType_INTS,
SetAttrTupleValueToProto))
SetAttrTupleValueToProto<0>))


OPERATOR_ONNX_CONVERT_DEFINE( OPERATOR_ONNX_CONVERT_DEFINE(
Conv2D, Conv, Conv2D, Conv,
OpNameInfo() OpNameInfo()
.Attr("dilation", "dilations", onnx::AttributeProto_AttributeType_INTS, SetAttrValueToProto<Int32Imm, 2>) .Attr("dilation", "dilations", onnx::AttributeProto_AttributeType_INTS, SetAttrValueToProto<Int32Imm, 2>)
.Attr("group", "group", onnx::AttributeProto_AttributeType_INT, SetAttrValueToProto<Int32Imm>) .Attr("group", "group", onnx::AttributeProto_AttributeType_INT, SetAttrValueToProto<Int32Imm>)
.Attr("kernel_size", "kernel_shape", onnx::AttributeProto_AttributeType_INTS, SetAttrTupleValueToProto)
.Attr("kernel_size", "kernel_shape", onnx::AttributeProto_AttributeType_INTS, SetAttrTupleValueToProto<0>)
.Attr("pad_mode", "auto_pad", onnx::AttributeProto_AttributeType_STRING, .Attr("pad_mode", "auto_pad", onnx::AttributeProto_AttributeType_STRING,
[](ValuePtr value, onnx::AttributeProto_AttributeType, onnx::AttributeProto* const attr_proto, [](ValuePtr value, onnx::AttributeProto_AttributeType, onnx::AttributeProto* const attr_proto,
const PrimitivePtr& prim) { const PrimitivePtr& prim) {


+ 1
- 1
mindspore/ccsrc/optimizer/irpass/incorporate_getitem.h View File

@@ -15,7 +15,7 @@
*/ */


#ifndef MINDSPORE_CCSRC_OPTIMIZER_IRPASS_INCORPORATE_GETITEM_H_ #ifndef MINDSPORE_CCSRC_OPTIMIZER_IRPASS_INCORPORATE_GETITEM_H_
#define MINDSPORE_CCSRC_OPTIMIZER_IRPASS_INCORPORATE_GETITEM_H__
#define MINDSPORE_CCSRC_OPTIMIZER_IRPASS_INCORPORATE_GETITEM_H_


#include <vector> #include <vector>
#include <algorithm> #include <algorithm>


+ 1
- 1
mindspore/ccsrc/parallel/ops_info/operator_info.cc View File

@@ -236,7 +236,7 @@ OperatorVector CreateMirrorOps(const std::string& group_name, size_t dev_num) {


OperatorName operator_name = MIRROR_OPERATOR; OperatorName operator_name = MIRROR_OPERATOR;
ValuePtr attr0_value = MakeValue(group_name); ValuePtr attr0_value = MakeValue(group_name);
ValuePtr attr1_value = MakeValue(dev_num);
ValuePtr attr1_value = MakeValue(SizeToInt(dev_num));
ValuePtr attr2_value = MakeValue(mean_flag); ValuePtr attr2_value = MakeValue(mean_flag);


Attr attr0 = std::make_pair(GROUP, attr0_value); Attr attr0 = std::make_pair(GROUP, attr0_value);


+ 1
- 0
mindspore/ccsrc/utils/log_adapter.h View File

@@ -22,6 +22,7 @@
#include <string> #include <string>
#include <sstream> #include <sstream>
#include <memory> #include <memory>
#include "./overload.h"
#include "./securec.h" #include "./securec.h"
#ifdef USE_GLOG #ifdef USE_GLOG
#include "glog/logging.h" #include "glog/logging.h"


Loading…
Cancel
Save