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

JIT: No longer using fallbacks for stride[0] > 1

Browse Source 
It was pretty messy.
This commit is contained in:
Philippe Tillet
2016-04-10 16:31:29 -04:00
parent 81139e0642
commit 1e439ad5bc
20 changed files with 5232 additions and 113 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
lib/jit/generation/reduce_2d.cpp
View File

@@ -313,7 +313,7 @@ std::vector<int_t> reduce_2d::input_sizes(expression_tree const & tree) const
return {shape[0], shape[1]};
}
void reduce_2d::enqueue(driver::CommandQueue & queue, driver::Program const & program, std::string const & suffix, base &, runtime::execution_handler const & control)
void reduce_2d::enqueue(driver::CommandQueue & queue, driver::Program const & program, std::string const & suffix, runtime::execution_handler const & control)
{
expression_tree const & tree = control.x();
std::vector<int_t> MN = input_sizes(tree);