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 69d16842..968dedf7 100644 --- a/ge/common/formats/format_transfers/format_transfer_fractal_z.cc +++ b/ge/common/formats/format_transfers/format_transfer_fractal_z.cc @@ -25,7 +25,7 @@ #include "framework/common/debug/ge_log.h" #include "framework/common/debug/log.h" #include "graph/utils/type_utils.h" - +#include "iostream" namespace ge { namespace formats { namespace { @@ -519,7 +519,8 @@ Status FormatTransferFractalZ::TransFormat(const TransArgs &args, TransResult &r return TransFormatNhwcToFz(args, result); } if ((args.src_format == FORMAT_HWCN) && (GetPrimaryFormat(args.dst_format) == FORMAT_FRACTAL_Z)) { - if (GetSubFormat(args.dst_format) >= 1) { + if (GetSubFormat(args.dst_format) > 1) { + std::cout<<"come in groups process"<= 1) { + if (GetSubFormat(dst_format) > 1) { return TransShapeHwcnToFzWithGroups(src_shape, data_type, dst_shape, GetSubFormat(dst_format)); } return TransShapeHwcnToFz(src_shape, data_type, dst_shape);