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

Fixed bug on marix-vector products with vectorization

Browse Source
This commit is contained in:
Philippe Tillet
2015-02-10 03:09:12 -05:00
parent a89f6d88be
commit 37fc98c532
3 changed files with 37 additions and 37 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
lib/backend/templates/mreduction.cpp
View File

@@ -81,7 +81,7 @@ std::string mreduction::generate_impl(unsigned int label, expressions_tuple cons
if(reduction==REDUCE_COLUMNS)
{
accessors["array2"] = data_type + " #namereg = " + vload(simd_width, "c*#stride1", "#pointer + r*#ld")+";";
accessors["repeat"] = "#scalartype #namereg = $VALUE{(c%#tuplearg0)*#stride, (r%#tuplearg1)*#stride};";
accessors["repeat"] = data_type + " #namereg = " + vload(simd_width, "(c%#tuplearg0)*#stride", "#pointer + (r%#tuplearg1)*#stride ")+";";
}
else
{
@@ -107,7 +107,7 @@ std::string mreduction::generate_impl(unsigned int label, expressions_tuple cons
{
std::map<std::string, std::string> accessors;
accessors["array2"] = str[a];
accessors["repeat"] = "#namereg";
accessors["repeat"] = str[a];
accessors["array0"] = "#namereg";
std::string value = elem->evaluate_recursive(LHS_NODE_TYPE, accessors);
if (elem->is_index_reduction())