[PYTHON][EXAMPLES] Added self-attention example using triton.ops.einsum

2020-01-21 16:45:04 -05:00
parent 78b98fb7cf
commit ce7a00674a
8 changed files with 227 additions and 47 deletions
--- a/lib/runtime/function.cc
+++ b/lib/runtime/function.cc
@@ -256,7 +256,6 @@ std::unique_ptr<driver::module> function::make_bin(ir::module &module, driver::c
    return std::unique_ptr<driver::module>();
  barriers.run(module);
  isel.visit(module, *llvm);
-
  // return binary
  std::unique_ptr<driver::module> res(driver::module::create(context, std::move(llvm)));
  // done