diff --git a/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_util.h b/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_util.h index b96fc46c66c..caabe162a9a 100644 --- a/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_util.h +++ b/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_util.h @@ -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 kOpNameToAicpuOpNameMap{ {kUpsampleNearest3DGrad, "UpsampleNearest3dGrad"}, {kNameRangeV2, "Range"}, {kReLUV3, "Relu"}, + {kFillV2, "Fill"}, {kUpsampleTrilinear3D, "UpsampleTrilinear3d"}, {kUpsampleTrilinear3DGrad, "UpsampleTrilinear3dGrad"}, {kStack, "Pack"}, diff --git a/mindspore/ccsrc/plugin/device/ascend/optimizer/mindir/aicpu_lib_select.cc b/mindspore/ccsrc/plugin/device/ascend/optimizer/mindir/aicpu_lib_select.cc index 9eb58dc963a..5b20867d819 100644 --- a/mindspore/ccsrc/plugin/device/ascend/optimizer/mindir/aicpu_lib_select.cc +++ b/mindspore/ccsrc/plugin/device/ascend/optimizer/mindir/aicpu_lib_select.cc @@ -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, diff --git a/mindspore/ccsrc/plugin/device/ascend/optimizer/mindir/reg_ascend_vm_op_adaptation_info.h b/mindspore/ccsrc/plugin/device/ascend/optimizer/mindir/reg_ascend_vm_op_adaptation_info.h index 5d4533d8ef4..f925839f059 100644 --- a/mindspore/ccsrc/plugin/device/ascend/optimizer/mindir/reg_ascend_vm_op_adaptation_info.h +++ b/mindspore/ccsrc/plugin/device/ascend/optimizer/mindir/reg_ascend_vm_op_adaptation_info.h @@ -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);