gaspersic/triton
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

[examples] debugging matrix multiplication code

Browse Source
This commit is contained in:
Philippe Tillet
2019-02-08 13:15:04 -05:00
parent 90c0474974
commit 937bc464a3
1 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
examples/matrix.cpp
View File

@@ -40,7 +40,7 @@ void test(fp32 *a, fp32 *b, fp32 *c, int32 M, int32 N, int32 K){\
fp32* pa[32, 8] = a + rx[:, newaxis] + rka[newaxis, :]*M;\ fp32* pa[32, 8] = a + rx[:, newaxis] + rka[newaxis, :]*M;\
fp32* pb[32, 8] = b + ry[:, newaxis] + rkb[newaxis, :]*K;\ fp32* pb[32, 8] = b + ry[:, newaxis] + rkb[newaxis, :]*K;\
fp32* pc[32, 32] = c + rx[:, newaxis] + ry[newaxis, :]*M;\ fp32* pc[32, 32] = c + rx[:, newaxis] + ry[newaxis, :]*M;\
for(k = K; k >= 0; k = k - 8){\ for(k = K; k > 0; k = k - 8){\
fp32 a[32, 8] = *pa;\ fp32 a[32, 8] = *pa;\
fp32 b[32, 8] = *pb;\ fp32 b[32, 8] = *pb;\
C = C + 1;\ C = C + 1;\
@@ -228,5 +228,8 @@ int main() {
// Write back // Write back
checkCudaErrors(cuMemcpyDtoH(c.data(), d_c, sizeof(numeric_t) * c.size())); checkCudaErrors(cuMemcpyDtoH(c.data(), d_c, sizeof(numeric_t) * c.size()));
for(size_t i = 0; i < M*N; i++)
if(c[i] == 32)
std::cout << i << " " << "success" << std::endl;
return 0; return 0;
} }