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

Backend: Fixed AXPY for shape=(1,x>1)

Browse Source
This commit is contained in:
Philippe Tillet
2015-07-01 11:48:01 -04:00
parent 9d0d50ba05
commit 5c720a5b54
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
lib/backend/templates/vaxpy.cpp
View File

@@ -103,8 +103,9 @@ vaxpy::vaxpy(unsigned int simd, unsigned int ls, unsigned int ng,
std::vector<int_t> vaxpy::input_sizes(expressions_tuple const & expressions) const
{
size4 shape = static_cast<array_expression const *>(expressions.data().front().get())->shape();
int_t size = static_cast<array_expression const *>(expressions.data().front().get())->shape()[0];
return tools::make_vector<int_t>() << size;
return tools::make_vector<int_t>() << std::max(shape[0], shape[1]);
}
void vaxpy::enqueue(driver::CommandQueue & queue, driver::Program & program, const char * suffix, base & fallback, controller<expressions_tuple> const & controller)