| @@ -289,10 +289,16 @@ Status TransFormatHwcnToFzWithGroups(const TransArgs &args, TransResult &result, | |||||
| int64_t c1_dim = cin_opt / cube_k; | int64_t c1_dim = cin_opt / cube_k; | ||||
| int64_t g_dim = DivCeil(groups, e_mult); | int64_t g_dim = DivCeil(groups, e_mult); | ||||
| int64_t dim_cin = cin_opt / cube_k; | int64_t dim_cin = cin_opt / cube_k; | ||||
| std::cout<<"cin_opt:"<<cin_opt<<std::endl; | |||||
| std::cout<<"cout_opt:"<<cout_opt<<std::endl; | |||||
| std::cout<<"c1_dim:"<<c1_dim<<std::endl; | |||||
| std::cout<<"g_dim:"<<g_dim<<std::endl; | |||||
| int64_t data_size = GetCubeSizeByDataType(args.src_data_type); | int64_t data_size = GetCubeSizeByDataType(args.src_data_type); | ||||
| std::cout<<"data_size:"<<data_size<<std::endl; | |||||
| int64_t size_output_data = | int64_t size_output_data = | ||||
| g_dim * kDim * dim_cin * h_dim * w_dim * cout_opt * cube_k * data_size; | g_dim * kDim * dim_cin * h_dim * w_dim * cout_opt * cube_k * data_size; | ||||
| GE_CHK_BOOL_EXEC_NOLOG(size_output_data != 0, result.length = static_cast<size_t>(size_output_data); | GE_CHK_BOOL_EXEC_NOLOG(size_output_data != 0, result.length = static_cast<size_t>(size_output_data); | ||||
| std::cout<<"size_output_data:"<<size_output_data<<std::endl; | |||||
| return SUCCESS;); | return SUCCESS;); | ||||
| errno_t ret = EOK; | errno_t ret = EOK; | ||||
| std::shared_ptr<uint8_t> dst(new (std::nothrow) uint8_t[size_output_data], std::default_delete<uint8_t[]>()); | std::shared_ptr<uint8_t> dst(new (std::nothrow) uint8_t[size_output_data], std::default_delete<uint8_t[]>()); | ||||