• Lei Wang's avatar
    [Enhancement] Introduce pass_configs parameter for kernel Caching (#452) · b1ba0cc8
    Lei Wang authored
    * [Enhancement] Introduce pass_configs parameter for kernel compilation
    
    * Added a new `pass_configs` parameter to the `tilelang.compile` function to allow for more flexible kernel compilation configurations.
    * Updated related classes and methods to accommodate the new parameter, ensuring compatibility across the codebase.
    * Enhanced the `torch_assert_close` function to include customizable tensor names for better debugging output.
    * Refactored input handling in example scripts to streamline the process of obtaining inputs for kernel execution.
    
    * lint fix
    b1ba0cc8
example_retnet.py 8.86 KB