NASP/cuda-samples
50
0
Fork 0
mirror of https://github.com/NVIDIA/cuda-samples.git synced 2025-08-23 23:35:31 +08:00
Code Issues Packages Projects Releases Wiki Activity

Merge branch 'shawnz_bug_fix' into 'master'

Browse Source 
Bug 5412815: Fix the issue of cudaTensorCoreGemm.cu

See merge request cuda-samples/cuda-samples!125
This commit is contained in:
Rob Armstrong 2025-07-28 10:18:27 -07:00
commit a5267b83a5
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
Samples/3_CUDA_Features/cudaTensorCoreGemm/cudaTensorCoreGemm.cu
View File

@ -224,7 +224,7 @@ __global__ void compute_gemm(const half *A, const half *B, const float *C, float
// there's no such tile, all warps in this CTA exit.
for (unsigned int block_pos = blockIdx.x;; block_pos += gridDim.x) {
const unsigned int block_tile_i = ((block_pos * BLOCK_ROW_TILES) / N_TILES) * (BLOCK_COL_TILES);
const unsigned int block_tile_j = (block_pos * BLOCK_COL_TILES) % N_TILES;
const unsigned int block_tile_j = (block_pos * BLOCK_ROW_TILES) % N_TILES;
// Stop when there are no more D matrix tiles to compute in this CTA.
if (block_tile_i >= M_TILES) {