Browse Source

fix the adapter of UnsortedSegmentProd, UnsortedSegmentMax and AccumulateNV2 problem for graphengine.

tags/v1.3.0
wangshuide2020 4 years ago
parent
commit
c39f64ed18
3 changed files with 7 additions and 3 deletions
  1. +1
    -1
      mindspore/ccsrc/transform/graph_ir/op_declare/elewise_calculation_ops_declare.cc
  2. +4
    -2
      mindspore/ccsrc/transform/graph_ir/op_declare/selection_ops_declare.cc
  3. +2
    -0
      mindspore/ccsrc/transform/graph_ir/op_declare/selection_ops_declare.h

+ 1
- 1
mindspore/ccsrc/transform/graph_ir/op_declare/elewise_calculation_ops_declare.cc View File

@@ -38,7 +38,7 @@ REG_ADPT_DESC(Add, prim::kPrimAdd->name(),
// AccumulateNV2
INPUT_MAP(AccumulateNV2) = EMPTY_INPUT_MAP;
DYN_INPUT_MAP(AccumulateNV2) = {{1, DYN_INPUT_DESC(x)}};
ATTR_MAP(AccumulateNV2) = {{"N", ATTR_DESC(N, AnyTraits<int64_t>())}};
ATTR_MAP(AccumulateNV2) = {{"n", ATTR_DESC(N, AnyTraits<int64_t>())}};
OUTPUT_MAP(AccumulateNV2) = {{0, OUTPUT_DESC(y)}};
REG_ADPT_DESC(AccumulateNV2, kNameAccumulateNV2, ADPT_DESC(AccumulateNV2))



+ 4
- 2
mindspore/ccsrc/transform/graph_ir/op_declare/selection_ops_declare.cc View File

@@ -160,13 +160,15 @@ REG_ADPT_DESC(UnsortedSegmentSumD, prim::kPrimUnsortedSegmentSum->name(), ADPT_D

// UnsortedSegmentProdD
INPUT_MAP(UnsortedSegmentProdD) = {{1, INPUT_DESC(x)}, {2, INPUT_DESC(segment_ids)}};
ATTR_MAP(UnsortedSegmentProdD) = {{"num_segments", ATTR_DESC(num_segments, AnyTraits<int64_t>())}};
INPUT_ATTR_MAP(UnsortedSegmentProdD) = {{3, ATTR_DESC(num_segments, AnyTraits<int64_t>())}};
ATTR_MAP(UnsortedSegmentProdD) = EMPTY_ATTR_MAP;
OUTPUT_MAP(UnsortedSegmentProdD) = {{0, OUTPUT_DESC(y)}};
REG_ADPT_DESC(UnsortedSegmentProdD, kNameUnsortedSegmentProdD, ADPT_DESC(UnsortedSegmentProdD))

// UnsortedSegmentMaxD
INPUT_MAP(UnsortedSegmentMaxD) = {{1, INPUT_DESC(x)}, {2, INPUT_DESC(segment_ids)}};
ATTR_MAP(UnsortedSegmentMaxD) = {{"num_segments", ATTR_DESC(num_segments, AnyTraits<int64_t>())}};
INPUT_ATTR_MAP(UnsortedSegmentMaxD) = {{3, ATTR_DESC(num_segments, AnyTraits<int64_t>())}};
ATTR_MAP(UnsortedSegmentMaxD) = EMPTY_ATTR_MAP;
OUTPUT_MAP(UnsortedSegmentMaxD) = {{0, OUTPUT_DESC(y)}};
REG_ADPT_DESC(UnsortedSegmentMaxD, kNameUnsortedSegmentMaxD, ADPT_DESC(UnsortedSegmentMaxD))



+ 2
- 0
mindspore/ccsrc/transform/graph_ir/op_declare/selection_ops_declare.h View File

@@ -57,9 +57,11 @@ DECLARE_OP_USE_INPUT_ATTR(UnsortedSegmentSumD)
DECLARE_OP_USE_OUTPUT(UnsortedSegmentSumD)

DECLARE_OP_ADAPTER(UnsortedSegmentProdD)
DECLARE_OP_USE_INPUT_ATTR(UnsortedSegmentProdD)
DECLARE_OP_USE_OUTPUT(UnsortedSegmentProdD)

DECLARE_OP_ADAPTER(UnsortedSegmentMaxD)
DECLARE_OP_USE_INPUT_ATTR(UnsortedSegmentMaxD)
DECLARE_OP_USE_OUTPUT(UnsortedSegmentMaxD)

DECLARE_OP_ADAPTER(UnsortedSegmentMin)


Loading…
Cancel
Save