diff --git a/python/triton/code_gen.py b/python/triton/code_gen.py
index 709ada423..35c097017 100644
--- a/python/triton/code_gen.py
+++ b/python/triton/code_gen.py
@@ -822,6 +822,7 @@ class Kernel:
             torch.float32: 'f32',
             torch.float64: 'f64',
             torch.bool: 'i1',
+            torch.uint8: 'u8',
             torch.int8: 'i8',
             torch.int16: 'i16',
             torch.int32: 'i32',