Skip to content

Commit d4af439

Browse files
authored
Use int8_t C++ type for bool kernel args (#7624)
We recently fixed this for XPU backend, maybe it will be useful for other backends. Signed-off-by: Anatoly Myachev <[email protected]>
1 parent c6dbfba commit d4af439

File tree

2 files changed

+4
-4
lines changed

2 files changed

+4
-4
lines changed

third_party/amd/backend/driver.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -153,12 +153,12 @@ def ty_to_cpp(ty):
153153
if ty[0] == '*':
154154
return "hipDeviceptr_t"
155155
return {
156-
"i1": "int32_t",
156+
"i1": "int8_t",
157157
"i8": "int8_t",
158158
"i16": "int16_t",
159159
"i32": "int32_t",
160160
"i64": "int64_t",
161-
"u1": "uint32_t",
161+
"u1": "uint8_t",
162162
"u8": "uint8_t",
163163
"u16": "uint16_t",
164164
"u32": "uint32_t",

third_party/nvidia/backend/driver.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -84,12 +84,12 @@ def ty_to_cpp(ty):
8484
if ty.startswith("tensordesc"):
8585
return "CUtensorMap"
8686
return {
87-
"i1": "int32_t",
87+
"i1": "int8_t",
8888
"i8": "int8_t",
8989
"i16": "int16_t",
9090
"i32": "int32_t",
9191
"i64": "int64_t",
92-
"u1": "uint32_t",
92+
"u1": "uint8_t",
9393
"u8": "uint8_t",
9494
"u16": "uint16_t",
9595
"u32": "uint32_t",

0 commit comments

Comments
 (0)