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

trying to remove interior logic

Browse Source
This commit is contained in:
Philippe Tillet
2019-07-16 18:47:50 -07:00
parent 5f6dd23fc2
commit ec24e1e7df
5 changed files with 20 additions and 30 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
lib/dnn/base.cpp
View File

@@ -51,7 +51,7 @@ void base::enqueue(driver::stream *stream, std::vector<driver::buffer *> args, b
jit->add_module(name_.c_str(), src.c_str(), best.params);
}
else {
jit->add_module(name_.c_str(), src.c_str(), {16, 4, 128, 16, 4, 128, 2, 2, 2, 2, 8, 16, 8, 1});
jit->add_module(name_.c_str(), src.c_str(), jit->get_valid(name_.c_str(), src.c_str()));
}
triton::driver::kernel* kernel = jit->get_function(name_.c_str());
clone->init_impl(stream, (triton::driver::cu_module*)kernel->module());