lukasplevac/cuda-samples
1
0
Fork 0
mirror of https://github.com/NVIDIA/cuda-samples.git synced 2025-04-10 18:22:11 +01:00
Code Issues Packages Projects Releases Wiki Activity

Merge fb9f77575a08918a682d7eb0781c6e5f1d4f0bf3 into 3e8f91d1a116060d3fedfe856f3721db970de030

Browse Source
This commit is contained in:
Tailing Yuan 2025-03-04 08:42:41 +08:00 committed by GitHub
commit 9a15c3d631
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
Samples/3_CUDA_Features/bf16TensorCoreGemm/bf16TensorCoreGemm.cu
View File

@ -592,8 +592,8 @@ __global__ void simple_wmma_bf16gemm(__nv_bfloat16 *a, __nv_bfloat16 *b, float *
int aCol = i;
int aRow = warpM * M;
int bCol = i;
int bRow = warpN * N;
int bCol = warpN * N;
int bRow = i;
// Bounds checking
if (aRow < m_ld && aCol < k_ld && bRow < k_ld && bCol < n_ld) {