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

bla

Browse Source
This commit is contained in:
Philippe Tillet
2019-02-15 22:03:09 -05:00
parent 5f5959dc6e
commit cf1a583dbf
6 changed files with 36 additions and 25 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
examples/matrix.cpp
View File

@@ -54,8 +54,8 @@ void test(fp32 *a, fp32 *b, fp32 *c, int32 M, int32 N, int32 K, int32 bound){\
int1 checkc1[16] = ryc < N;\
int1 checkc[16, 16] = checkc0[:, newaxis] && checkc1[newaxis, :];\
for(k = K; k > 0; k = k - 8){\
int1 sanitya[16, 8] = (k >= bound);\
int1 sanityb[16, 8] = (k >= bound);\
int1 sanitya[16, 8] = (k > 16);\
int1 sanityb[16, 8] = (k > 16);\
C = dot(a, b, C);\
pa = pa + 8*M;\
pb = pb + 8*K;\
@@ -236,7 +236,7 @@ int main() {
// execute machine code
// Allocate buffers
typedef float numeric_t;
size_t M = 128, N = 128, K = 128;
size_t M = 32, N = 32, K = 32;
size_t bound = 8;
std::vector<numeric_t> c(M*N);
std::vector<numeric_t> rc(M*N);