.

triton-lang · Feb 10, 2024 · f94c4ca · f94c4ca
1 parent 9899d83
commit f94c4ca
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/python/triton/compiler/code_generator.py b/python/triton/compiler/code_generator.py
@@ -1172,6 +1172,7 @@ def str_to_ty(name):
         "bf16": language.bfloat16,
         "fp32": language.float32,
         "fp64": language.float64,
+        "i1": language.int1,
         "i8": language.int8,
         "i16": language.int16,
         "i32": language.int32,

diff --git a/third_party/nvidia/backend/driver.py b/third_party/nvidia/backend/driver.py
@@ -91,6 +91,7 @@ def ty_to_cpp(ty):
     if ty[0] == '*':
         return "CUdeviceptr"
     return {
+        "i1": "int32_t",
         "i8": "int8_t",
         "i16": "int16_t",
         "i32": "int32_t",