Skip to content

Commit 24dd8db

Browse files
author
limingming
committed
gpu
1 parent 93986dc commit 24dd8db

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

matrixMulCUBLAS.h

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -369,7 +369,7 @@ checkCudaErrors(cublasDgemm(handle, CUBLAS_OP_N, CUBLAS_OP_N, matrix_size.uiWB,
369369
////////////////////////////////////////////////////////////////////////////////
370370
// Program main
371371
////////////////////////////////////////////////////////////////////////////////
372-
int bytomcall(int argc, char **argv,int8_t aaaa[][256],int8_t bbbb[][256])
372+
int bytomcall(int num, char **argu,int8_t aaaa[][256],int8_t bbbb[][256])
373373
{
374374
printf("[Matrix Multiply CUBLAS] - Starting...\n");
375375

@@ -381,9 +381,9 @@ int bytomcall(int argc, char **argv,int8_t aaaa[][256],int8_t bbbb[][256])
381381
matrix_sizeui.uiHB=256;
382382
matrix_sizeui.uiWC=256;
383383
matrix_sizeui.uiHC=256;
384-
initializeCUDA(argc, argv, devID, sizeMult, matrix_size);
384+
initializeCUDA(num, argu, devID, sizeMult, matrix_size);
385385

386-
int matrix_result = matrixMultiply(argc, argv, devID, matrix_size,aaaa,bbbb);
386+
int matrix_result = matrixMultiply(num, argu, devID, matrix_size,aaaa,bbbb);
387387

388388
return matrix_result;
389389
}

0 commit comments

Comments
 (0)