1. 18 Aug, 2023 1 commit
  2. 17 Aug, 2023 2 commits
    • Li Zhang's avatar
    • Chen Xin's avatar
      Support windows platform (#209) · 4c9959f6
      Chen Xin authored
      * __PRETTY_FUNCTION__
      
      * CASE_K
      
      * uint
      
      * remove not
      
      * HALF_FLT_MAX
      
      * struct init
      
      * port utils
      
      * better build pthread-win32
      
      * port kernels
      
      * port utils/gemm_test
      
      * hide windows header
      
      * port models
      
      * port examples && triton_backend && unittests
      
      * update build readme
      
      * fix lint
      
      * fix lint
      
      * fix lint
      
      * fix lint
      
      * fix lint
      
      * fix build
      
      * fix build
      
      * cmake version
      
      * fix typos
      
      * update ci
      
      * port kernels/gemm_s_f16
      
      * update ci
      
      * fix ci
      
      * use cudaStreamSynchronize instead of volatile check
      
      * remove gettimeofday
      
      * remove pthread-win32
      
      * remove dirent.h
      
      * update pre-commit
      
      * update
      
      * remove todo
      
      * fix include
      
      * fix build
      
      * fix build
      
      * fix build ci
      
      * fix github action trigger
      
      * update README
      
      * fix linux-build ci
      
      * remove windows folder
      
      * fix lint
      
      * update readme
      4c9959f6
  3. 15 Aug, 2023 1 commit
  4. 14 Aug, 2023 2 commits
  5. 03 Aug, 2023 1 commit
  6. 31 Jul, 2023 2 commits
  7. 27 Jul, 2023 1 commit
    • Chen Xin's avatar
      Add manylinux builder (#164) · b9004712
      Chen Xin authored
      
      
      * update builder
      
      * remove root permission
      
      * update readme
      
      * update setup.py
      
      * add install cuda 12.1 script
      
      * use generate.sh
      
      * add nccl to install_requires
      
      * update README.md
      
      * fix lint
      
      * update setup.py
      
      ---------
      Co-authored-by: default avatarchenxin <chenxin@pjlab.org.cn>
      b9004712
  8. 25 Jul, 2023 1 commit
  9. 24 Jul, 2023 1 commit
  10. 21 Jul, 2023 1 commit
  11. 18 Jul, 2023 1 commit
    • q.yao's avatar
      Tensor Parallel python api (#82) · 7cbfe2ea
      q.yao authored
      * wip
      
      * profile disable tp
      
      * fix profile
      
      * lint
      
      * fix dlpack
      
      * remove comment
      
      * add tp flag
      
      * add session len check
      
      * add eos
      
      * remove tp and session len inputs
      
      * warp tokenizer
      
      * multithread load weight
      
      * update profile
      
      * refactor tokenizer
      
      * remove pre/post process
      
      * remove mpi4py requirement
      
      * remove
      
      * remove bind
      
      * remove mpi requirement
      
      * check backend_tokenizer
      7cbfe2ea
  12. 17 Jul, 2023 1 commit
  13. 06 Jul, 2023 2 commits
  14. 05 Jul, 2023 2 commits
    • pppppM's avatar
      [Feature] Stats Quantization Parameters for KV Cache (#45) · 3fff964d
      pppppM authored
      * add cal qparams
      
      * support offload inference
      
      * add collect funtions (mod,weight)
      
      * stats kv scales
      
      * update init
      
      * add user guide
      
      * fix hints
      
      * fix comments & support turbomind format
      
      * update user guide
      
      * fix slice kv cache error & support pileval dataset (used in llm-awq)
      
      * fix wrong num heads slice
      
      * update default dataset
      
      * fix conflict
      
      * fix hints
      
      * fix hints
      
      * add gitignore
      3fff964d
    • q.yao's avatar
      Python ffi (#34) · 4fd6e710
      q.yao authored
      
      
      * wip
      
      * wip
      
      * example finish
      
      * fix include and namespace
      
      * wtf
      
      * install lib
      
      * batchize
      
      * update cmake install
      
      * multithread
      
      * fix comment
      
      * fix
      
      * add mmengine
      
      * bind llamamodel
      
      ---------
      Co-authored-by: default avatargrimoire <yaoqian@pjlab.org.cn>
      4fd6e710
  15. 04 Jul, 2023 1 commit
  16. 03 Jul, 2023 2 commits
  17. 01 Jul, 2023 4 commits
  18. 28 Jun, 2023 2 commits
    • tpoisonooo's avatar
      feat(src): add kv cache int8 quantization (#22) · cc93136e
      tpoisonooo authored
      * feat(src): add int8 and compile passed
      
      * feat(kernels): fix
      
      * feat(llama): update kernel
      
      * feat(src): add debug
      
      * fix(kernel): k_cache use int8_t pointer
      
      * style(llama): clean code
      
      * feat(deploy.py): revert to enable fmha
      
      * style(LlamaV2): clean code
      
      * feat(deploy.py): add default quant policy
      cc93136e
    • Li Zhang's avatar
      fix-gemm-tuning (#24) · 4d42a781
      Li Zhang authored
      4d42a781
  19. 26 Jun, 2023 1 commit
  20. 24 Jun, 2023 1 commit
  21. 22 Jun, 2023 1 commit
  22. 21 Jun, 2023 1 commit
    • q.yao's avatar
      support fmha (#9) · 6c7d9992
      q.yao authored
      * support fmha
      
      * update sm by cudaarch
      
      * update ldscript path
      
      * clang-format
      
      * clang-format
      
      ---------
      6c7d9992
  23. 20 Jun, 2023 1 commit