gaspersic/triton
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

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

Browse Source 
Co-authored-by: Philippe Tillet <Phil.Tillet@gmail.com>
This commit is contained in:
Jason Ansel
2022-05-12 13:07:39 -07:00
committed by GitHub
parent 0835a4fb05
commit d954a05989
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
python/triton/code_gen.py
View File

@@ -822,6 +822,7 @@ class Kernel:
torch.float32: 'f32', torch.float32: 'f32',
torch.float64: 'f64', torch.float64: 'f64',
torch.bool: 'i1', torch.bool: 'i1',
torch.uint8: 'u8',
torch.int8: 'i8', torch.int8: 'i8',
torch.int16: 'i16', torch.int16: 'i16',
torch.int32: 'i32', torch.int32: 'i32',