squall/cutlass
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

Update basic_gemm.cu

Browse Source 
fix the matrix malloc size
This commit is contained in:
Haicheng Wu 2021-06-15 09:08:36 -04:00 committed by GitHub
parent 6a1064093f
commit 856d4db3fb
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
examples/00_basic_gemm/basic_gemm.cu
View File

@ -187,7 +187,7 @@ cudaError_t InitializeMatrix(float *matrix, int rows, int columns, int seed = 0)
cudaError_t AllocateMatrix(float **matrix, int rows, int columns, int seed = 0) { cudaError_t AllocateMatrix(float **matrix, int rows, int columns, int seed = 0) {
cudaError_t result; cudaError_t result;
size_t sizeof_matrix = sizeof(float) * columns; size_t sizeof_matrix = sizeof(float) * rows * columns;
// Allocate device memory. // Allocate device memory.
result = cudaMalloc(reinterpret_cast<void **>(matrix), sizeof_matrix); result = cudaMalloc(reinterpret_cast<void **>(matrix), sizeof_matrix);