Thanks to visit codestin.com
Credit goes to github.com

Skip to content

Commit ff817e6

Browse files
haijieghaijieg
committed
Update API docs
1 parent 9d357fe commit ff817e6

File tree

4 files changed

+235
-178
lines changed

4 files changed

+235
-178
lines changed

‎docs/source/conf.py‎

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -36,6 +36,7 @@
3636

3737
# Autodoc settings
3838
autodoc_member_order = 'bysource'
39+
autodoc_typehints = 'none'
3940
toc_object_entries = False # Don't include object entries in the TOC
4041

4142
# -- Options for HTML output -------------------------------------------------

‎src/cuda/tile/_ir/ops.py‎

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -3229,8 +3229,9 @@ def generate_bytecode(self, ctx: BytecodeContext) -> bc.Value:
32293229

32303230

32313231
def arange(size: int, dtype: DType) -> Var:
3232-
res_ty = make_tile_ty(dtype, (size,))
3233-
return add_operation(TileArange, res_ty)
3232+
res_ty = make_tile_ty(datatype.default_int_type, (size,))
3233+
res = add_operation(TileArange, res_ty)
3234+
return astype(res, dtype)
32343235

32353236

32363237
@impl(ct.arange)

0 commit comments

Comments
 (0)