[BACKEND] Add bf16 & tf32 mma supports (on A100) (#426)

2022-01-12 02:20:31 +08:00
parent efdabe6073
commit 94a2e10fe5
17 changed files with 717 additions and 263 deletions
--- a/lib/driver/llvm.cc
+++ b/lib/driver/llvm.cc
@@ -121,6 +121,7 @@ std::string llir_to_ptx(llvm::Module* module, int cc, int version){
  init_llvm();
  // verify and store llvm
  llvm::legacy::PassManager pm;
+  // module->print(llvm::outs(), nullptr);
  pm.add(llvm::createVerifierPass());
  pm.run(*module);
  // create machine