Commit Graph
10 Commits
Author SHA1 Message Date
Kuris 4f844000e3 [Fix] Fix missing not rewrite in frontend (#1348) 2025-11-26 19:27:43 +08:00
Kuris 71b73e185a [Refactor] Disable strided buffer load inside tvm (#1301) (#1332) 2025-11-25 12:32:48 +08:00
KurisandLei Wang bf90a5f58c [Fix] Fix frame scope error in T.macro (#1308)
* [Fix] Fix #1307 by adding macro inside function

* fix lint error

* add comments and fix lint error

* Remove debug print from enter_frame method

Removed debug print statement from enter_frame method.

---------

Co-authored-by: Lei Wang <34334180+LeiWang1999@users.noreply.github.com>
2025-11-21 20:27:14 +08:00
Kuris 2426090fdb [Fix] Remove unused let_bindings_ in CodeGenC to fix #1300 (#1305)
* [Feat] add missing support of uint32x2

* [Feat] Add `T.Ref` annotation and tests

* fix lint error

* minor update for error message on twice decl

* Remove unused let_bindings_ in CodeGenC to fix #1300
2025-11-21 17:04:52 +08:00
Kuris dd7fdb8ee9 [Feat] add support for passing reference in T.Var annotation (#1291) 2025-11-20 17:33:35 +08:00
Kuris 041d4a06b5 [Refactor] add support for numpy dtype conversion (#1255)
* add typing stub for tir.ir

* remove idents

* minor update

* [Refactor] add numpy conversion for dtype

* fix lint error

* remove unused np.float_ in dtype conversion

* fix type in np.int_

* fix typo

* minor fix

* remove debug files
2025-11-17 13:06:23 +08:00
Kuris 5eb30a4f3b [Language] Add missing while statement (#1254)
* add typing stub for tir.ir

* remove idents

* minor update

* [Language] Add missing while statement

* add test
2025-11-14 13:31:20 +08:00
777881e1e8 [Feat] Add support for T.serial with step and negative step (#1188)
* [Feature] Support serial for with step

* add more tests

* fix

* Enhance trip count validation in SerialForWithStep to ensure non-zero step values and prevent undefined behavior. Added error handling for zero step values and improved logging for non-constant steps.

* Update builder.py

* fix lint error

---------

Co-authored-by: Zhiwen Mo <zm125@ic.ac.uk>
Co-authored-by: Lei Wang <34334180+LeiWang1999@users.noreply.github.com>
2025-11-06 17:34:13 +08:00
Kurisu 055f850017 [Feat] Add swap like grammar in tuple assignment (#1185)
* [Feat] add 2 phase binding to allow swap two var

* Minor update tvm dtype constructor

* fix lint error
2025-11-05 12:03:35 +08:00
5f202fe5c1 [Language] Initial version of tilelang frontend v2 (#1120)
* tilelang frontend v2

* syntax sugar: defining a local var by annotation

* [Refactor] fix type linting warning like `T.float32`

* Add tl.local_var_init for new tl.float32

* allow passing default argument as function annotation

* allow default arguments as annotation

* fix lint error

* minor fix

* [Refactor] refactor tilelang.jit and tilelang.autotune

* minor fix

* minor fix

* minor fix

* fix metal get function name

* add par_compile impl and tests

* Type consistency on tvm datatype
1. isinstance(tl.float32, tvm.DataType) == True
2. Allow `tl.float32` as function annotations
3. Allow `tl.float32` as argument to be passed to `tl.alloc` or other functions

* fix lint error

* add more warning in frontend

* update tvm version

* Minor fix on tvm_ffi annotations

* add document and examples

* fix lint error

* Simplify index calculations in example_chunk_o_bwd.py

Refactor index calculations for dg_last_fragment assignment.

* minor fix

* lint fix

---------

Co-authored-by: Lei Wang <leiwang1999@outlook.com>
Co-authored-by: Lei Wang <34334180+LeiWang1999@users.noreply.github.com>
2025-11-03 18:15:56 +08:00