• Lei Wang's avatar
    [Refactor] Set default log level from waning into info (#132) · 9ba96f19
    Lei Wang authored
    * Change default log level from WARNING to INFO in TileLang initialization
    
    * Refactor Flash Attention Variable-Length MHA Example with Cython Backend Support
    
    - Update `example_mha_fwd_varlen.py` to use Cython backend for kernel compilation
    - Remove unused imports and simplify function signature
    - Modify `flashattn` function to handle max sequence length as a separate argument
    - Update kernel call to include max sequence length parameter
    - Improve code readability and remove commented-out code
    - Add print statement to confirm successful assertion
    
    * Refactor code formatting in TileLang lowering and example files
    
    - Improve line breaks and code formatting in `lower.py`, `wrapper.py`, and `tensor.py`
    - Simplify line breaks and reduce unnecessary whitespace
    - Enhance code readability by adjusting indentation and line breaks
    - Update example MHA forward pass script with cleaner tensor initialization
    9ba96f19
example_mha_fwd_varlen.py 17.8 KB