From 94ce6aa80feb1d077fd278fc7c74879cd56d9917 Mon Sep 17 00:00:00 2001
From: Philippe Tillet <ptillet@g.harvard.edu>
Date: Wed, 21 Jul 2021 14:12:41 -0700
Subject: [PATCH] [DRIVER] Added support for CUDA 11.4 (#135)

---
 lib/driver/module.cc | 1 +
 1 file changed, 1 insertion(+)

diff --git a/lib/driver/module.cc b/lib/driver/module.cc
index aba374c7a..c31e3cca4 100755
--- 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) {