Hu Chunming / vpt_ascend_arm

Commit f6cb983a21fa1845d081f90d5087a260ec3c219c

Authored by Hu Chunming
1 parent 4356373c

修改demo

Inline Side-by-side
Showing 1 changed file with 16 additions and 14 deletions
src/demo/demo.cpp
  View file @f6cb983
... ... @@ -1230,18 +1230,20 @@ void test_gpu(int gpuID){
1230 1230  
1231 1231 // }
1232 1232  
1233   - createTask(handle, algor_vec2, 0, false);
1234   - createTask(handle, algor_vec2, 2, false);
1235   - createTask(handle, algor_vec2, 0, false);
1236   - createTask(handle, algor_vec2, 2, false);
1237   - createTask(handle, algor_vec2, 0, false);
1238   - createTask(handle, algor_vec2, 2, false);
1239   - createTask(handle, algor_vec2, 0, false);
1240   - createTask(handle, algor_vec2, 2, false);
1241   - createTask(handle, algor_vec2, 0, false);
1242   - createTask(handle, algor_vec2, 2, false);
1243   - createTask(handle, algor_vec2, 0, false);
1244   - createTask(handle, algor_vec2, 2, false);
  1233 + // createTask(handle, algor_vec2, 0, false);
  1234 + // createTask(handle, algor_vec2, 2, false);
  1235 + // createTask(handle, algor_vec2, 0, false);
  1236 + // createTask(handle, algor_vec2, 2, false);
  1237 + // createTask(handle, algor_vec2, 0, false);
  1238 + // createTask(handle, algor_vec2, 2, false);
  1239 + // createTask(handle, algor_vec2, 0, false);
  1240 + // createTask(handle, algor_vec2, 2, false);
  1241 + // createTask(handle, algor_vec2, 0, false);
  1242 + // createTask(handle, algor_vec2, 2, false);
  1243 + // createTask(handle, algor_vec2, 0, false);
  1244 + // createTask(handle, algor_vec2, 2, false);
  1245 +
  1246 + createTask(handle, algor_vec2, 4, false);
1245 1247  
1246 1248 while (getchar() != 'q');
1247 1249  
... ... @@ -1387,12 +1389,12 @@ int main(int argc, char *argv[]) {
1387 1389 // int repeat_num = atoi(argv[3]);
1388 1390 // int gpuID = atoi(argv[4]);
1389 1391  
1390   - // test_gpu(0);
  1392 + test_gpu(0);
1391 1393 // test_gpu(1);
1392 1394 // test_gpu(2);
1393 1395 // test_gpu(3);
1394 1396  
1395   - test_dvpp28181(0);
  1397 + // test_dvpp28181(0);
1396 1398  
1397 1399 printf("Done.\n");
1398 1400  
... ...