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 1002 case 27:
1003 1003 tparam.ipc_url = "/data/share/data/Street_4k_265.mp4";
1004 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 1011 default:
1006 1012 tparam.ipc_url = "/opt/share/data/Street.uvf";
1007 1013 break;
... ... @@ -1244,32 +1250,38 @@ void test_gpu(int gpuID){
1244 1250 // createTask(handle, algor_vec2, 0, false);
1245 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 1278 char ch = 'a';
1253 1279 while (ch != 'q') {
1254 1280 ch = getchar();
1255 1281 switch (ch)
1256 1282 {
1257 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 1285 break;
1274 1286 case 'c':
1275 1287 close_all_task(handle);
... ... @@ -1278,12 +1290,14 @@ void test_gpu(int gpuID){
1278 1290 break;
1279 1291 }
1280 1292  
1281   - }*/
  1293 + }
1282 1294  
1283 1295 // finish_task(handle, (char*)task_id.data(), 0);
1284 1296  
1285 1297 // finish_task(handle, (char*)task_id1.data(), 0);
1286 1298  
  1299 + while (getchar() != 'q');
  1300 +
1287 1301 tsl_aiplatform_release(&handle);
1288 1302 }
1289 1303  
... ... @@ -1410,12 +1424,12 @@ int main(int argc, char *argv[]) {
1410 1424 // int repeat_num = atoi(argv[3]);
1411 1425 // int gpuID = atoi(argv[4]);
1412 1426  
1413   - // test_gpu(0);
  1427 + test_gpu(0);
1414 1428 // test_gpu(1);
1415 1429 // test_gpu(2);
1416 1430 // test_gpu(3);
1417 1431  
1418   - test_dvpp28181(0);
  1432 + // test_dvpp28181(0);
1419 1433  
1420 1434 printf("Done.\n");
1421 1435  
... ...