• Kurisu's avatar
    [Language] Initial version of tilelang frontend v2 (#1120) · 5f202fe5
    Kurisu authored
    
    
    * 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: default avatarLei Wang <leiwang1999@outlook.com>
    Co-authored-by: default avatarLei Wang <34334180+LeiWang1999@users.noreply.github.com>
    5f202fe5
test_tilelang_language_let.py 647 Bytes