[PYTHON][EINSUM] Now handling reduction sizes that are not a multiple of
TK
This commit is contained in:
committed by
Philippe Tillet
parent
fa4ec7ea65
commit
3816f2f259
@@ -255,6 +255,7 @@ std::unique_ptr<driver::module> function::make_bin(ir::module &module, driver::c
|
||||
if(allocation.allocated_size() > context->device()->max_shared_memory())
|
||||
return std::unique_ptr<driver::module>();
|
||||
barriers.run(module);
|
||||
// ir::print(module, std::cout);
|
||||
isel.visit(module, *llvm);
|
||||
// return binary
|
||||
std::unique_ptr<driver::module> res(driver::module::create(context, std::move(llvm)));
|
||||
|
Reference in New Issue
Block a user