[DRIVER] Added support for CUDA 11.4 (#135)

2021-07-21 14:12:41 -07:00
parent 01276b5153
commit 94ce6aa80f
1 changed files with 1 additions and 0 deletions
--- a/lib/driver/module.cc
+++ b/lib/driver/module.cc
@@ -211,6 +211,7 @@ static std::map<int, int> vptx = {
  {11010, 71},
  {11020, 72},
  {11030, 73},
+  {11040, 74}
 };

 std::string cu_module::compile_llvm_module(llvm::Module* module, driver::device* device) {