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

Kernels: merged start1, start2 and stride1, stride2 into start and stride for matrices

Browse Source
This commit is contained in:
Philippe Tillet
2015-08-10 22:45:48 -07:00
parent 5365b1331f
commit 963867574f
7 changed files with 26 additions and 35 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
lib/kernels/templates/tools/arguments.hpp
View File

@@ -22,7 +22,7 @@ inline std::string generate_arguments(std::string const & data_type, driver::Dev
process(stream, PARENT_NODE_TYPE, { {"array0", kwglobal + " #scalartype* #pointer, " + _size_t + " #start,"},
{"host_scalar", "#scalartype #name,"},
{"array1", kwglobal + " " + data_type + "* #pointer, " + _size_t + " #start, " + _size_t + " #stride,"},
{"array2", kwglobal + " " + data_type + "* #pointer, " + _size_t + " #ld, " + _size_t + " #start1, " + _size_t + " #start2, " + _size_t + " #stride1, " + _size_t + " #stride2,"},
{"array2", kwglobal + " " + data_type + "* #pointer, " + _size_t + " #ld, " + _size_t + " #start, " + _size_t + " #stride, "},
{"tuple4", "#scalartype #name0, #scalartype #name1, #scalartype #name2, #scalartype #name3,"}}
, expressions, mappings);
@@ -81,11 +81,9 @@ public:
}
else
{
kernel_.setSizeArg(current_arg_++, a->ld());
kernel_.setSizeArg(current_arg_++, a->start()[0]);
kernel_.setSizeArg(current_arg_++, a->start()[1]);
kernel_.setSizeArg(current_arg_++, a->ld()*a->stride()[1]);
kernel_.setSizeArg(current_arg_++, a->start()[0] + a->start()[1]*a->ld());
kernel_.setSizeArg(current_arg_++, a->stride()[0]);
kernel_.setSizeArg(current_arg_++, a->stride()[1]);
}
}
}