| 332 | } |
| 333 | |
| 334 | int comp_res(node_t out_1, node_t out_2, int data_type) |
| 335 | { |
| 336 | tensor_t out1_tensor = get_node_output_tensor(out_1, 0); |
| 337 | tensor_t out2_tensor = get_node_output_tensor(out_2, 0); |
| 338 | |
| 339 | int out1_size = get_tensor_buffer_size(out1_tensor); |
| 340 | int out2_size = get_tensor_buffer_size(out2_tensor) / sizeof(float); |
| 341 | |
| 342 | int8_t* out1_buf = ( int8_t* )get_tensor_buffer(out1_tensor); |
| 343 | int8_t* out2_buf = ( int8_t* )get_tensor_buffer(out2_tensor); |
| 344 | |
| 345 | if(data_type == TENGINE_DT_FP32) |
| 346 | { |
| 347 | int num = out1_size / sizeof(float); |
| 348 | if(num != out2_size) |
| 349 | { |
| 350 | std::cout << "---------------------fp32 not match--------------------------\n"; |
| 351 | } |
| 352 | |
| 353 | float* pOut1 = ( float* )out1_buf; |
| 354 | float* pOut2 = ( float* )out2_buf; |
| 355 | for(int ii = 0; ii < num; ++ii) |
| 356 | { |
| 357 | if(!comp_fp32(pOut1[ii], pOut2[ii])) |
| 358 | { |
| 359 | std::cout << "---------------------fp32 not match--------------------------\n"; |
| 360 | return 1; |
| 361 | } |
| 362 | } |
| 363 | } |
| 364 | else if(data_type == TENGINE_DT_FP16) |
| 365 | { |
| 366 | int num = out1_size / sizeof(__fp16); |
| 367 | if(num != out2_size) |
| 368 | { |
| 369 | std::cout << "---------------------fp16 not match--------------------------\n"; |
| 370 | } |
| 371 | __fp16* pOut1 = ( __fp16* )out1_buf; |
| 372 | float* pOut2 = ( float* )out2_buf; |
| 373 | for(int ii = 0; ii < num; ++ii) |
| 374 | { |
| 375 | if(!comp_fp16(pOut1[ii], pOut2[ii])) |
| 376 | { |
| 377 | std::cout << "---------------------fp16 not match--------------------------\n"; |
| 378 | return 1; |
| 379 | } |
| 380 | } |
| 381 | } |
| 382 | else if(data_type == TENGINE_DT_INT8) |
| 383 | { |
| 384 | int num = out1_size; |
| 385 | if(num != out2_size) |
| 386 | { |
| 387 | std::cout << "---------------------int8 not match--------------------------\n"; |
| 388 | } |
| 389 | int8_t* pOut1 = out1_buf; |
| 390 | float* pOut2 = ( float* )out2_buf; |
| 391 | for(int ii = 0; ii < num; ++ii) |
no test coverage detected