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

ugh

Browse Source
This commit is contained in:
Philippe Tillet
2019-07-16 12:59:27 -07:00
parent f50d7a420a
commit 7d1797cd32
6 changed files with 27 additions and 28 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
lib/runtime/jit.cpp
View File

@@ -37,20 +37,24 @@ void loop_nest(std::vector<size_t> const & ranges,
size_t D = ranges.size();
std::vector<size_t> values(D, 0);
// thread pools
nbsdx::concurrent::thread_pool pool(nthreads);
// nbsdx::concurrent::thread_pool pool(nthreads);
// Start with innermost loop
size_t i = D - 1;
size_t current = 0;
// size_t current = 0;
while(true){
//Execute function
pool.add_job([values, &f](){ f(values); });
// f(values);
// pool.add_job([values, &f](){ f(values); });
f(values);
//Increment counters
while(values[i]++ == ranges[i] - 1){
if(i == 0)
return;
values[i--] = 0;
}
// if(current++ >= 1024){
// current = 0;
// pool.join_all();
// }
i = D - 1;
}
}
@@ -128,7 +132,6 @@ std::vector<unsigned> jit::get_valid(const char *name, const char *src) {
ranges.push_back(mp->get_space());
// iterate over parameters
std::vector<unsigned> result;
size_t nthreads = 1;
loop_nest<unsigned>(ranges, [&](const std::vector<unsigned> params){
if(!result.empty())
return;
@@ -146,7 +149,7 @@ std::vector<unsigned> jit::get_valid(const char *name, const char *src) {
if(!errors.empty())
return;
result = params;
}, nthreads);
}, 1);
if(result.empty())
throw std::runtime_error("couldn't find valid parameters");
return result;