1. 27 May, 2024 1 commit
  2. 12 Jan, 2024 1 commit
  3. 20 Dec, 2023 1 commit
  4. 10 Nov, 2023 1 commit
    • Li Zhang's avatar
      TurboMind 2 (#590) · ab1767cf
      Li Zhang authored
      * refresh decoder attention kernel
      
      * block-level kv cache
      
      * `BlockManager` & `SequenceManager`
      
      * update
      
      * update
      
      * update
      
      * update
      
      * rename
      
      * GQA support
      
      * fix context length
      
      * GQA dispatch
      
      * kv8
      
      * tune
      
      * async stream cb
      
      * nvtx
      
      * config parsing
      
      * debug
      
      * optimize output cost
      
      * split-k decoding
      
      * minor
      
      * truncate `session_len` by available blocks
      
      * minor
      
      * license
      
      * fix
      
      * dispatch `cp.async`
      
      * fix linking
      
      * fix
      
      * fix deadlock
      
      * guard input length
      
      * correct start offset
      
      * fix prefill chunking
      
      * fix `cache_block_seq_len` param passing
      
      * fix `block_size` fmtstr
      
      * fix output tokens
      
      * fix batch resizing
      
      * fix masking of finished sequences
      
      * add debug util
      
      * free unused block early
      
      * add ntk scaling and logn scaling
      
      * cmake flags
      
      * fix typo
      
      * w4a16 for sm75
      
      * fix msvc build
      
      * fix msvc build
      
      * fix block verification
      
      * fix msvc build
      
      * use `std::shuffle`
      
      * fix lint
      
      * fix lint
      
      * fix lint
      
      * clear incoming buffer
      
      * clear finished requests
      
      * fix batch initialization
      
      * fix typo
      
      * fix typo
      
      * fix comparison
      ab1767cf
  5. 14 Aug, 2023 1 commit
    • Li Zhang's avatar
      [Feature] Blazing fast W4A16 inference (#202) · c3290cad
      Li Zhang authored
      * add w4a16
      
      * fix `deploy.py`
      
      * add doc
      
      * add w4a16 kernels
      
      * fuse w1/w3 & bugfixes
      
      * fix typo
      
      * python
      
      * guard sm75/80 features
      
      * add missing header
      
      * refactor
      
      * qkvo bias
      
      * update cost model
      
      * fix lint
      
      * update `deploy.py`
      c3290cad
  6. 31 Jul, 2023 1 commit
  7. 01 Jul, 2023 1 commit
  8. 20 Jun, 2023 1 commit