"...layout/git@developer.sourcefind.cn:wangsen/mineru.git" did not exist on "ebfab424612a35018c37a0848f04e18738bb429f"
  • Yu Cheng's avatar
    [CI][Test] Add test cases for tilelang kernel FlashAttention (#54) · bedab1a0
    Yu Cheng authored
    * [Dev] Add FlashDecoding example
    
    * [CI][Test] Add test cases for tilelang kernel convolution
    
    * [CI][Test] Add test cases for tilelang kernel FlashAttention
    
    * Reduce the number of stages to ensure the shared memory allocation is valid
    
    * Temporarily remove the dim128 case
    
    * lint
    
    * update einops in requirements-dev.txt
    
    * update einops in requirements-test.txt
    
    * remove einops in requirements-dev.txt
    bedab1a0
example_mha_inference.py 14.1 KB