From d954a059893c2446a58ab73e0d898dee3c124f49 Mon Sep 17 00:00:00 2001
From: Jason Ansel <jansel@jansel.net>
Date: Thu, 12 May 2022 13:07:39 -0700
Subject: [PATCH] [FRONTEND] Handle torch.uint8 args (#513)

Co-authored-by: Philippe Tillet <Phil.Tillet@gmail.com>
---
 python/triton/code_gen.py | 1 +
 1 file changed, 1 insertion(+)

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',