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

BLAS: Removed nasty temporary in GEMM...

Browse Source
This commit is contained in:
Philippe Tillet
2016-09-30 00:28:17 -04:00
parent 7210098e1a
commit 8bf7344681
3 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
include/isaac/jit/syntax/expression/expression.h
View File

@@ -81,6 +81,7 @@ struct array_holder
{ {
int_t start; int_t start;
handle_t handle; handle_t handle;
array_base* base;
}; };
class expression_tree class expression_tree

1
lib/jit/syntax/expression/expression.cpp
View File

@@ -47,6 +47,7 @@ expression_tree::node::node(value_scalar const & x) : type(VALUE_SCALAR_TYPE), d
expression_tree::node::node(array_base const & x) : type(DENSE_ARRAY_TYPE), dtype(x.dtype()), shape(x.shape()) expression_tree::node::node(array_base const & x) : type(DENSE_ARRAY_TYPE), dtype(x.dtype()), shape(x.shape())
{ {
array.start = x.start(); array.start = x.start();
array.base = (array_base*)&x;
driver::Buffer::handle_type const & h = x.data().handle(); driver::Buffer::handle_type const & h = x.data().handle();
switch(h.backend()){ switch(h.backend()){
case driver::OPENCL: array.handle.cl = h.cl(); break; case driver::OPENCL: array.handle.cl = h.cl(); break;

2
lib/jit/syntax/expression/preset.cpp
View File

@@ -119,7 +119,7 @@ matrix_product::args matrix_product::check(expression_tree::data_type const & tr
} }
if(result.C == NULL) if(result.C == NULL)
result.C = &left; result.C = &left;
else if(result.C != &left) else if(result.C->array.base != left.array.base)
result.C = NULL; result.C = NULL;
return result; return result;
} }