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

more work

Browse Source
This commit is contained in:
Philippe Tillet
2019-10-26 15:10:19 -04:00
parent 76adcb755a
commit 655f43fb5b
5 changed files with 46 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
lib/codegen/selection/generator.cc
View File

@@ -187,6 +187,7 @@ generator::generator(analysis::axes *a_axes,
void generator::visit_value(ir::value* v) {
std::cout << "visiting " << typeid(*v).name() << std::endl;
if(!seen_.insert(v).second)
return;
// create machine tile
@@ -559,8 +560,8 @@ void generator::visit_hmma_dot(ir::dot_inst* dot, shared_tile *TA, shared_tile *
bool is_a_trans = is_trans(dot->get_operand(0));
bool is_b_trans = is_trans(dot->get_operand(1));
bool is_a_row = is_a_trans ^ (ord_a[ord_a.size() - 2] == 1);
bool is_b_row = is_b_trans ^ (ord_b[ord_b.size() - 2] == 1);
bool is_a_row = is_a_trans ^ (ord_a[0] == 1);
bool is_b_row = is_b_trans ^ (ord_b[0] == 1);
Value *offset_a_i = hmma->offset_a_i_;

1
lib/codegen/selection/machine_value.cc
View File

@@ -1,4 +1,5 @@
#include <numeric>
#include <iostream>
#include "llvm/IR/IRBuilder.h"
#include "triton/codegen/selection/machine_value.h"