Nigel
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Modify fillv2 tbe registration

This commit is contained in:
panzhihui 2023-02-09 21:10:09 +08:00
parent b9c8db3b80
commit 61fde60c99
3 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_util.h
View File

@ -183,6 +183,7 @@ constexpr auto kSign = "Sign";
constexpr auto kFmax = "Fmax";
constexpr auto kGLU = "GLU";
constexpr auto kFmin = "Fmin";
constexpr auto kFillV2 = "FillV2";
constexpr auto kArgmax = "Argmax";
constexpr auto kArgmin = "Argmin";
constexpr auto kRange = "Range";
@ -315,6 +316,7 @@ const std::map<std::string, std::string> kOpNameToAicpuOpNameMap{
{kUpsampleNearest3DGrad, "UpsampleNearest3dGrad"},
{kNameRangeV2, "Range"},
{kReLUV3, "Relu"},
{kFillV2, "Fill"},
{kUpsampleTrilinear3D, "UpsampleTrilinear3d"},
{kUpsampleTrilinear3DGrad, "UpsampleTrilinear3dGrad"},
{kStack, "Pack"},

2
mindspore/ccsrc/plugin/device/ascend/optimizer/mindir/aicpu_lib_select.cc
View File

@ -72,7 +72,7 @@ const AnfNodePtr AICpuLibSelectPass::Process(const FuncGraphPtr &graph, const An
mindspore::kCSRSparseMatrixToDenseOpName,
mindspore::kCSRSparseMatrixToSparseTensorOpName,
mindspore::kDataFormatVecPermuteOpName,
mindspore::kFillOpName,
mindspore::kFillV2OpName,
mindspore::kLogMatrixDeterminantOpName,
mindspore::kMatrixSolveLsOpName,
mindspore::kMedianOpName,

2
mindspore/ccsrc/plugin/device/ascend/optimizer/mindir/reg_ascend_vm_op_adaptation_info.h
View File

@ -285,7 +285,7 @@ REG_ASCEND_VM_OP_ADAPTATION_INFO(kExpandDimsOpName).set_target_op_name(kExpandDi
REG_ASCEND_VM_OP_ADAPTATION_INFO(kFillOpName).set_target_op_name(kFillDOpName);
REG_ASCEND_VM_OP_ADAPTATION_INFO(kFillV2OpName).set_backend_op_name(kFillOpName).set_need_tbe_check_supported(true);
REG_ASCEND_VM_OP_ADAPTATION_INFO(kFillV2OpName).set_backend_op_name(kFillOpName);
// In hisi code, first check dynamic impl in GatherV2
REG_ASCEND_VM_OP_ADAPTATION_INFO(kGatherOpName).set_backend_op_name(kGatherV2OpName);