1. 03 Mar, 2025 1 commit
    • Sayak Paul's avatar
      [Tests] Remove more encode prompts tests (#10942) · 7513162b
      Sayak Paul authored
      * fix-copies went uncaught it seems.
      
      * remove more unneeded encode_prompt() tests
      
      * Revert "fix-copies went uncaught it seems."
      
      This reverts commit eefb302791172a4fb8ef008e400f94878de2c6c9.
      
      * empty
      7513162b
  2. 02 Sep, 2024 1 commit
  3. 06 Aug, 2024 1 commit
  4. 24 Jul, 2024 1 commit
    • Sayak Paul's avatar
      [Core] fix QKV fusion for attention (#8829) · 50d21f7c
      Sayak Paul authored
      * start debugging the problem,
      
      * start
      
      * fix
      
      * fix
      
      * fix imports.
      
      * handle hunyuan
      
      * remove residuals.
      
      * add a check for making sure there's appropriate procs.
      
      * add more rigor to the tests.
      
      * fix test
      
      * remove redundant check
      
      * fix-copies
      
      * move check_qkv_fusion_matches_attn_procs_length and check_qkv_fusion_processors_exist.
      50d21f7c
  5. 12 Jun, 2024 1 commit