Skip to content

Commit e1f4420

Browse files
committed
Fix type promotion of u16 and u64
Signed-off-by: Jay Gu <jagu@nvidia.com>
1 parent 2d163fc commit e1f4420

File tree

3 files changed

+3
-1
lines changed

3 files changed

+3
-1
lines changed

changelog.d/fix-dtype-promotion.md

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1 @@
1+
- Fix a type promotion bug where the promoted dtype of uint16 and uint64 was incorrectly set to uint32.

scripts/check_license.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -10,6 +10,7 @@
1010
".svg",
1111
"uv.lock",
1212
"LICENSES/",
13+
"changelog.d/",
1314
]
1415

1516
max_lines_to_check = 10

src/cuda/tile/_datatype.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -340,7 +340,7 @@ class _DTypePromotionImpl:
340340
# b1, u8, u16, u32, u64, i8, i16, i32, i64, f16, f32, f64, bf, tf, f8e4m3fn, f8e5m2
341341
[b1, u8, u16, u32, u64, i8, i16, i32, i64, f16, f32, f64, bf, na, na, na], # b1
342342
[u8, u8, u16, u32, u64, na, na, na, na, f16, f32, f64, bf, na, na, na], # u8
343-
[u16, u16, u16, u32, u32, na, na, na, na, f16, f32, f64, bf, na, na, na], # u16
343+
[u16, u16, u16, u32, u64, na, na, na, na, f16, f32, f64, bf, na, na, na], # u16
344344
[u32, u32, u32, u32, u64, na, na, na, na, f16, f32, f64, bf, na, na, na], # u32
345345
[u64, u64, u64, u64, u64, na, na, na, na, f16, f32, f64, bf, na, na, na], # u64
346346
[i8, na, na, na, na, i8, i16, i32, i64, f16, f32, f64, bf, na, na, na], # i8

0 commit comments

Comments
 (0)