Hu Chunming / vpt_ascend_arm

Commit 60108b8b6f9f022ab67cef64348f7abbeee5aa4e

Authored by Hu Chunming
1 parent caef8aa2

补充测试案例

Inline Side-by-side
Showing 1 changed file with 35 additions and 21 deletions
src/demo/demo.cpp
  View file @60108b8
@@ -1002,6 +1002,12 @@ string createTask(void *handle, std::vector<algorithm_type_t> algor_vec, int gi, @@ -1002,6 +1002,12 @@ string createTask(void *handle, std::vector<algorithm_type_t> algor_vec, int gi,
1002 case 27: 1002 case 27:
1003 tparam.ipc_url = "/data/share/data/Street_4k_265.mp4"; 1003 tparam.ipc_url = "/data/share/data/Street_4k_265.mp4";
1004 break; 1004 break;
  1005 + case 28:
  1006 + tparam.ipc_url = "http://192.168.60.179:10016/公安局老桥头_CVR15F89410_1465819864_1B.mp4";
  1007 + break;
  1008 + case 29:
  1009 + tparam.ipc_url = "http://192.168.60.179:10016/不带头盔2.mp4";
  1010 + break;
1005 default: 1011 default:
1006 tparam.ipc_url = "/opt/share/data/Street.uvf"; 1012 tparam.ipc_url = "/opt/share/data/Street.uvf";
1007 break; 1013 break;
@@ -1244,32 +1250,38 @@ void test_gpu(int gpuID){ @@ -1244,32 +1250,38 @@ void test_gpu(int gpuID){
1244 // createTask(handle, algor_vec2, 0, false); 1250 // createTask(handle, algor_vec2, 0, false);
1245 // createTask(handle, algor_vec2, 2, false); 1251 // createTask(handle, algor_vec2, 2, false);
1246 1252
1247 - createTask(handle, algor_vec2, 4, false); 1253 + // createTask(handle, algor_vec2, 4, false);
  1254 +
  1255 + // int task_index = 0;
  1256 + // while (true)
  1257 + // {
  1258 + // int task_count = count_running_task(handle);
  1259 + // if (task_count < 7)
  1260 + // {
  1261 + // // if (task_index > 7)
  1262 + // // {
  1263 + // // task_index = 0;
  1264 + // // break;
  1265 + // // }
  1266 +
  1267 + // createTask(handle, algor_vec2, 28, false);
  1268 + // task_index++;
  1269 + // createTask(handle, algor_vec2, 29, false);
  1270 + // task_index++;
  1271 + // }
  1272 +
  1273 + // std::this_thread::sleep_for(std::chrono::seconds(5));
  1274 + // }
1248 1275
1249 - while (getchar() != 'q'); 1276 +
1250 1277
1251 - /*  
1252 char ch = 'a'; 1278 char ch = 'a';
1253 while (ch != 'q') { 1279 while (ch != 'q') {
1254 ch = getchar(); 1280 ch = getchar();
1255 switch (ch) 1281 switch (ch)
1256 { 1282 {
1257 case 'a': 1283 case 'a':
1258 - createTask(handle, algor_vec2, 4, false);  
1259 - createTask(handle, algor_vec2, 5, false);  
1260 - createTask(handle, algor_vec2, 6, false);  
1261 - createTask(handle, algor_vec2, 7, false);  
1262 - createTask(handle, algor_vec2, 8, false);  
1263 - createTask(handle, algor_vec2, 9, false);  
1264 - createTask(handle, algor_vec2, 10, false);  
1265 - // createTask(handle, algor_vec2, 11, false);  
1266 - // createTask(handle, algor_vec2, 12, false);  
1267 - // createTask(handle, algor_vec2, 13, false);  
1268 - // createTask(handle, algor_vec2, 14, false);  
1269 - // createTask(handle, algor_vec2, 15, false);  
1270 - // createTask(handle, algor_vec2, 16, false);  
1271 - // createTask(handle, algor_vec2, 17, false);  
1272 - // createTask(handle, algor_vec2, 18, false); 1284 + createTask(handle, algor_vec2, 28, false);
1273 break; 1285 break;
1274 case 'c': 1286 case 'c':
1275 close_all_task(handle); 1287 close_all_task(handle);
@@ -1278,12 +1290,14 @@ void test_gpu(int gpuID){ @@ -1278,12 +1290,14 @@ void test_gpu(int gpuID){
1278 break; 1290 break;
1279 } 1291 }
1280 1292
1281 - }*/ 1293 + }
1282 1294
1283 // finish_task(handle, (char*)task_id.data(), 0); 1295 // finish_task(handle, (char*)task_id.data(), 0);
1284 1296
1285 // finish_task(handle, (char*)task_id1.data(), 0); 1297 // finish_task(handle, (char*)task_id1.data(), 0);
1286 1298
  1299 + while (getchar() != 'q');
  1300 +
1287 tsl_aiplatform_release(&handle); 1301 tsl_aiplatform_release(&handle);
1288 } 1302 }
1289 1303
@@ -1410,12 +1424,12 @@ int main(int argc, char *argv[]) { @@ -1410,12 +1424,12 @@ int main(int argc, char *argv[]) {
1410 // int repeat_num = atoi(argv[3]); 1424 // int repeat_num = atoi(argv[3]);
1411 // int gpuID = atoi(argv[4]); 1425 // int gpuID = atoi(argv[4]);
1412 1426
1413 - // test_gpu(0); 1427 + test_gpu(0);
1414 // test_gpu(1); 1428 // test_gpu(1);
1415 // test_gpu(2); 1429 // test_gpu(2);
1416 // test_gpu(3); 1430 // test_gpu(3);
1417 1431
1418 - test_dvpp28181(0); 1432 + // test_dvpp28181(0);
1419 1433
1420 printf("Done.\n"); 1434 printf("Done.\n");
1421 1435