diff options
Diffstat (limited to 'cuda-kernels/v16p_kernel.cu')
| -rw-r--r-- | cuda-kernels/v16p_kernel.cu | 9 |
1 files changed, 5 insertions, 4 deletions
diff --git a/cuda-kernels/v16p_kernel.cu b/cuda-kernels/v16p_kernel.cu index 31a1460..864597d 100644 --- a/cuda-kernels/v16p_kernel.cu +++ b/cuda-kernels/v16p_kernel.cu @@ -1,5 +1,6 @@ #include <stdio.h> #include <curand.h> +#include <stdlib.h> // Define some error checking macros. #define cudaErrCheck(stat) { cudaErrCheck_((stat), __FILE__, __LINE__); } @@ -205,7 +206,7 @@ int main(int argc, char* argv[]) { cudaEvent_t startWMMA; cudaEvent_t stopWMMA; - + srand(time(NULL)); cudaErrCheck(cudaEventCreate(&startWMMA)); cudaErrCheck(cudaEventCreate(&stopWMMA)); @@ -231,7 +232,7 @@ int main(int argc, char* argv[]) { printf("a_int32\n"); for(int m=0;m<MATRIX_M;m++){ for(int n=0;n<MATRIX_K;n++){ - a_host_wmma[m*MATRIX_K+n]=(m*MATRIX_K+n)%4; + a_host_wmma[m*MATRIX_K+n]= random()%5; printf("%d ",a_host_wmma[m*MATRIX_K+n]); } printf(";\n"); @@ -240,7 +241,7 @@ int main(int argc, char* argv[]) { printf("b_int32\n"); for(int m=0;m<MATRIX_K;m++){ for(int n=0;n<MATRIX_N;n++){ - b_host_wmma[m*MATRIX_N+n]=(m*MATRIX_N+n)%256; + b_host_wmma[m*MATRIX_N+n]=random()%32; printf("%d ",b_host_wmma[m*MATRIX_N+n]); } printf(";\n"); @@ -249,7 +250,7 @@ int main(int argc, char* argv[]) { printf("c_int32\n"); for(int m=0;m<MATRIX_M;m++){ for(int n=0;n<MATRIX_N;n++){ - c_host_wmma[m*MATRIX_N+n]=(m*MATRIX_N+n)%8; + c_host_wmma[m*MATRIX_N+n]=random()%32; d_cal_host_wmma[m*MATRIX_N+n]=0; printf("%d ",c_host_wmma[m*MATRIX_N+n]); } |