From d275cf64cd5ea479d19fbcc97e06567ce932e7fa Mon Sep 17 00:00:00 2001 From: qiuzhongya Date: Tue, 15 Nov 2022 11:02:10 +0800 Subject: [PATCH] support for acl --- .../graph_ir/op_declare/selection_ops_declare.cc | 14 ++++++++------ .../graph_ir/op_declare/selection_ops_declare.h | 5 ++--- 2 files changed, 10 insertions(+), 9 deletions(-) diff --git a/mindspore/ccsrc/transform/graph_ir/op_declare/selection_ops_declare.cc b/mindspore/ccsrc/transform/graph_ir/op_declare/selection_ops_declare.cc index cdeacffa2c9..b0429d6b615 100644 --- a/mindspore/ccsrc/transform/graph_ir/op_declare/selection_ops_declare.cc +++ b/mindspore/ccsrc/transform/graph_ir/op_declare/selection_ops_declare.cc @@ -179,16 +179,18 @@ REG_ADPT_DESC(StridedSliceV2, kNameStridedSliceV2, ADPT_DESC(StridedSliceV2)) // UnsortedSegmentSum INPUT_MAP(UnsortedSegmentSum) = {{1, INPUT_DESC(x)}, {2, INPUT_DESC(segment_ids)}, {3, INPUT_DESC(num_segments)}}; +ATTR_INPUT_MAP(UnsortedSegmentSum) = {{"num_segments", 3}}; ATTR_MAP(UnsortedSegmentSum) = EMPTY_ATTR_MAP; OUTPUT_MAP(UnsortedSegmentSum) = {{0, OUTPUT_DESC(y)}}; +REG_ADPT_DESC(UnsortedSegmentSumD, prim::kPrimUnsortedSegmentSumD->name(), ADPT_DESC(UnsortedSegmentSum)) REG_ADPT_DESC(UnsortedSegmentSum, prim::kPrimUnsortedSegmentSum->name(), ADPT_DESC(UnsortedSegmentSum)) -// UnsortedSegmentProdD -INPUT_MAP(UnsortedSegmentProdD) = {{1, INPUT_DESC(x)}, {2, INPUT_DESC(segment_ids)}}; -INPUT_ATTR_MAP(UnsortedSegmentProdD) = {{3, ATTR_DESC(num_segments, AnyTraits())}}; -ATTR_MAP(UnsortedSegmentProdD) = EMPTY_ATTR_MAP; -OUTPUT_MAP(UnsortedSegmentProdD) = {{0, OUTPUT_DESC(y)}}; -REG_ADPT_DESC(UnsortedSegmentProdD, kNameUnsortedSegmentProdD, ADPT_DESC(UnsortedSegmentProdD)) +// UnsortedSegmentProd +INPUT_MAP(UnsortedSegmentProd) = {{1, INPUT_DESC(x)}, {2, INPUT_DESC(segment_ids)}, {3, INPUT_DESC(num_segments)}}; +ATTR_INPUT_MAP(UnsortedSegmentProd) = {{"num_segments", 3}}; +ATTR_MAP(UnsortedSegmentProd) = EMPTY_ATTR_MAP; +OUTPUT_MAP(UnsortedSegmentProd) = {{0, OUTPUT_DESC(y)}}; +REG_ADPT_DESC(UnsortedSegmentProd, kNameUnsortedSegmentProdD, ADPT_DESC(UnsortedSegmentProd)) // UnsortedSegmentMaxD INPUT_MAP(UnsortedSegmentMaxD) = {{1, INPUT_DESC(x)}, {2, INPUT_DESC(segment_ids)}}; diff --git a/mindspore/ccsrc/transform/graph_ir/op_declare/selection_ops_declare.h b/mindspore/ccsrc/transform/graph_ir/op_declare/selection_ops_declare.h index f0ba41c5116..7feba27f573 100644 --- a/mindspore/ccsrc/transform/graph_ir/op_declare/selection_ops_declare.h +++ b/mindspore/ccsrc/transform/graph_ir/op_declare/selection_ops_declare.h @@ -58,9 +58,8 @@ DECLARE_OP_USE_OUTPUT(StridedSliceV2) DECLARE_OP_ADAPTER(UnsortedSegmentSum) DECLARE_OP_USE_OUTPUT(UnsortedSegmentSum) -DECLARE_OP_ADAPTER(UnsortedSegmentProdD) -DECLARE_OP_USE_INPUT_ATTR(UnsortedSegmentProdD) -DECLARE_OP_USE_OUTPUT(UnsortedSegmentProdD) +DECLARE_OP_ADAPTER(UnsortedSegmentProd) +DECLARE_OP_USE_OUTPUT(UnsortedSegmentProd) DECLARE_OP_ADAPTER(UnsortedSegmentMaxD) DECLARE_OP_USE_INPUT_ATTR(UnsortedSegmentMaxD)