Merge branch 'jg/fix-arange' into 'main'

haijieg · haijieg · commit d28a06e147bd · 2025-11-25T17:19:12.000-08:00
Avoid typecast for arange when dtype is int

See merge request dl/tileir/cutile-python!51
diff --git a/src/cuda/tile/_ir/ops.py b/src/cuda/tile/_ir/ops.py
@@ -3229,7 +3229,10 @@ def generate_bytecode(self, ctx: BytecodeContext) -> bc.Value:
 
 
 def arange(size: int, dtype: DType) -> Var:
-    res_ty = make_tile_ty(datatype.default_int_type, (size,))
+    if datatype.is_integral(dtype):
+        res_ty = make_tile_ty(dtype, (size,))
+    else:
+        res_ty = make_tile_ty(datatype.default_int_type, (size,))
     res = add_operation(TileArange, res_ty)
     return astype(res, dtype)
 
