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(Squeeze, Squeeze,
OpNameInfo().Attr("axis", "axes", onnx::AttributeProto_AttributeType_INTS,
SetAttrTupleValueToProto))
SetAttrTupleValueToProto<0>))

OPERATOR_ONNX_CONVERT_DEFINE(
Conv2D, Conv,
OpNameInfo()
.Attr("dilation", "dilations", onnx::AttributeProto_AttributeType_INTS, SetAttrValueToProto<Int32Imm, 2>)
.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,
[](ValuePtr value, onnx::AttributeProto_AttributeType, onnx::AttributeProto* const attr_proto,
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_
#define MINDSPORE_CCSRC_OPTIMIZER_IRPASS_INCORPORATE_GETITEM_H__
#define MINDSPORE_CCSRC_OPTIMIZER_IRPASS_INCORPORATE_GETITEM_H_

#include <vector>
#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;
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);

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 <sstream>
#include <memory>
#include "./overload.h"
#include "./securec.h"
#ifdef USE_GLOG
#include "glog/logging.h"


Loading…
Cancel
Save