[FRONTEND] Handle torch.uint8 args (#513)

Co-authored-by: Philippe Tillet <Phil.Tillet@gmail.com>
2022-05-12 13:07:39 -07:00
parent 0835a4fb05
commit d954a05989
1 changed files with 1 additions and 0 deletions
--- 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',