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

[syntax tree] fixed bugs in control flow lowering

Browse Source
This commit is contained in:
Philippe Tillet
2019-02-17 21:35:03 -05:00
parent cf1a583dbf
commit f3094a512b
8 changed files with 201 additions and 35 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
examples/matrix.cpp
View File

@@ -5,6 +5,7 @@
#include "ast/ast.h"
#include "ir/context.h"
#include "ir/module.h"
#include "ir/print.h"
#include "codegen/selection.h"
#include "codegen/tune.h"
#include "codegen/shared_copy.h"
@@ -54,13 +55,15 @@ 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 > 16);\
int1 sanityb[16, 8] = (k > 16);\
int1 checka[16, 8] = (k > 8);\
int1 checkb[16, 8] = (k > 8);\
C = dot(a, b, C);\
pa = pa + 8*M;\
pb = pb + 8*K;\
@sanitya a = *pa;\
@sanityb b = *pb;\
@checka a = *pa;\
@checkb b = *pb;\
if(k <= 8){\
}\
}\
@checkc *pc = C;\
}\
@@ -167,6 +170,8 @@ int main() {
llvm::LLVMContext llvm_context;
llvm::Module llvm_module("test", llvm_context);
tdl::ir::print(module, std::cout);
// create passes
tdl::codegen::place_shared_copy shared;
tdl::codegen::buffer_info_pass buffer_info;