"driver/device_implicit_gemm_convolution_nchw_kcsr.cuh" did not exist on "913afaeb5d5ec0732d7b5a3da468ffd07609538e"
  • Cunxiao Ni's avatar
    [Example] Update examples to use @tilelang.jit (#597) · 3db18726
    Cunxiao Ni authored
    
    
    * [Example] Update kernel compilation in examples to use @tilelang.jit
    
    - Refactored multiple examples to eliminate the use of `tilelang.compile` for kernel creation, directly invoking the functions instead.
    - Added `@tilelang.jit` decorators with appropriate output indices to enhance performance and maintainability.
    - Improved code clarity by simplifying the kernel invocation process across various examples, ensuring consistency in how kernels are defined and executed.
    
    * format
    
    * Update example_tilelang_sparse_gqa_decode_varlen_indice.py
    
    * Update example_dequant_gemm_fine_grained.py
    
    * Update example_gemm_autotune.py
    
    ---------
    Co-authored-by: default avatarLei Wang <34334180+LeiWang1999@users.noreply.github.com>
    3db18726
example_mla_decode_paged.py 18.5 KB