| @@ -65,7 +65,7 @@ Status CheckArgsForHwcnToFrazlstm(const TransArgs &args) { | |||||
| GELOGE(PARAM_INVALID, "Failed to check src shape %s", ShapeToString(args.src_shape).c_str()); | GELOGE(PARAM_INVALID, "Failed to check src shape %s", ShapeToString(args.src_shape).c_str()); | ||||
| return PARAM_INVALID; | return PARAM_INVALID; | ||||
| } | } | ||||
| if (!CheckShapeValid(args.dst_shape, kFracZnLstmDimsNum)) { | |||||
| if (!CheckShapeValid(args.dst_shape, kFracZDimsNum)) { | |||||
| GELOGE(PARAM_INVALID, "Failed to check dst shape %s", ShapeToString(args.dst_shape).c_str()); | GELOGE(PARAM_INVALID, "Failed to check dst shape %s", ShapeToString(args.dst_shape).c_str()); | ||||
| return PARAM_INVALID; | return PARAM_INVALID; | ||||
| } | } | ||||
| @@ -96,6 +96,10 @@ Status GetDstDataAfterTrans(const TransArgs &args, TransResult &result, const in | |||||
| } | } | ||||
| auto ret = memcpy_s(dst.get(), static_cast<size_t>(total_size), args.data, static_cast<size_t>(total_size)); | auto ret = memcpy_s(dst.get(), static_cast<size_t>(total_size), args.data, static_cast<size_t>(total_size)); | ||||
| if (ret != EOK) { | |||||
| GELOGE(INTERNAL_ERROR, "Failed to copy data ==="); | |||||
| return INTERNAL_ERROR; | |||||
| } | |||||
| result.data = dst; | result.data = dst; | ||||
| result.length = static_cast<size_t>(total_size); | result.length = static_cast<size_t>(total_size); | ||||
| return SUCCESS; | return SUCCESS; | ||||