diff --git a/ge/common/formats/format_transfers/format_transfer_fractal_z.cc b/ge/common/formats/format_transfers/format_transfer_fractal_z.cc index 8b733d53..15312fad 100644 --- a/ge/common/formats/format_transfers/format_transfer_fractal_z.cc +++ b/ge/common/formats/format_transfers/format_transfer_fractal_z.cc @@ -85,7 +85,7 @@ Status TransShapeToFz(int64_t n, int64_t c, int64_t h, int64_t w, DataType data_ } Status TransShapeToFzWithGroups(int64_t n, int64_t c, int64_t h, int64_t w, DataType data_type, std::vector &dst_shape - , args.groups) { + , int64_t groups) { auto c0 = GetCubeSizeByDataType(data_type); if (c0 < 0) { return ACL_ERROR_GE_TRANSSHAPE_DATATYPE_INVALID; @@ -100,7 +100,7 @@ Status TransShapeToFzWithGroups(int64_t n, int64_t c, int64_t h, int64_t w, Data int64_t cout_opt = Ceil(e_mult * cout_ori, kCubeN) * kCubeN; int64_t c1_dim = cin_opt / cube_k; int64_t g_dim = Ceil(groups, e_mult); - auto n1 = Ceil(n, 16); + auto n1 = Ceil(n, kCubeN); dst_shape.clear(); dst_shape.push_back(g_dim * c1_dim * h * w); @@ -142,7 +142,7 @@ Status TransShapeHwcnToFz(const std::vector &src_shape, DataType data_t } TransShapeHwcnToFzWithGroups(const std::vector &src_shape, DataType data_type, std::vector &dst_shape -, groups){ +, int64_t groups){ if (!CheckShapeValid(src_shape, kHwcnDimsNum)) { return ACL_ERROR_GE_TRANSSHAPE_SHAPE_INVALID; } @@ -257,7 +257,7 @@ Status TransFormatFromNchwToFz(const TransArgs &args, TransResult &result) { return SUCCESS; } -Status TransFormatHwcnToFzWithGroups(const TransArgs &args, TransResult &result, groups){ +Status TransFormatHwcnToFzWithGroups(const TransArgs &args, TransResult &result,int64_t groups){ int64_t h_dim = args.src_shape[kHwcnH]; int64_t w_dim = args.src_shape[kHwcnW]; int64_t c_dim = args.src_shape[kHwcnC]; @@ -525,7 +525,7 @@ Status FormatTransferFractalZ::TransFormat(const TransArgs &args, TransResult &r } Status FormatTransferFractalZ::TransShape(Format src_format, const std::vector &src_shape, DataType data_type, - Format dst_format, std::vector &dst_shape, groups){ + Format dst_format, std::vector &dst_shape,int64_t groups){ if (CheckDataTypeSupport(data_type) != SUCCESS) { return ACL_ERROR_GE_TRANSSHAPE_DATATYPE_INVALID; }