update for cuda-11.4 (#1329)

Co-authored-by: Jeff Rasley <jerasley@microsoft.com>
microsoft · Aug 30, 2021 · b9ece25 · b9ece25
1 parent 336dd08
commit b9ece25
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/op_builder/builder.py b/op_builder/builder.py
@@ -70,7 +70,8 @@ def get_default_compute_capatabilities():
     11: ["11.0",
          "11.1",
          "11.2",
-         "11.3"],
+         "11.3",
+         "11.4"],
 }