mirror of
https://github.com/hpcaitech/ColossalAI.git
synced 2025-05-12 10:27:44 +00:00
f5c84af0b0
* halfway * fix cross-PP-stage position id length diff bug * fix typo * fix typo * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci * unified cross entropy func for all shardformer models * remove redundant lines * add basic ring attn; debug cross entropy * fwd bwd logic complete * fwd bwd logic complete; add experimental triton rescale * precision tests passed * precision tests passed * fix typos and remove misc files * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci * add sp_mode to benchmark; fix varlen interface * update softmax_lse shape by new interface * change tester name * remove buffer clone; support packed seq layout * add varlen tests * fix typo * all tests passed * add dkv_group; fix mask * remove debug statements --------- Co-authored-by: Edenzzzz <wtan45@wisc.edu> Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
41 lines
1.1 KiB
YAML
41 lines
1.1 KiB
YAML
repos:
|
|
|
|
- repo: https://github.com/PyCQA/autoflake
|
|
rev: v2.3.1
|
|
hooks:
|
|
- id: autoflake
|
|
name: autoflake (python)
|
|
args: ['--in-place', '--remove-unused-variables', '--remove-all-unused-imports', '--ignore-init-module-imports']
|
|
|
|
- repo: https://github.com/pycqa/isort
|
|
rev: 5.13.2
|
|
hooks:
|
|
- id: isort
|
|
name: sort all imports (python)
|
|
args: ["--profile", "black"] # avoid conflict with black
|
|
|
|
- repo: https://github.com/psf/black-pre-commit-mirror
|
|
rev: 24.8.0
|
|
hooks:
|
|
- id: black
|
|
name: black formatter
|
|
args: ['--line-length=120', '--target-version=py37', '--target-version=py38', '--target-version=py39','--target-version=py310']
|
|
|
|
- repo: https://github.com/pre-commit/mirrors-clang-format
|
|
rev: v18.1.8
|
|
hooks:
|
|
- id: clang-format
|
|
name: clang formatter
|
|
types_or: [c++, c]
|
|
|
|
- repo: https://github.com/pre-commit/pre-commit-hooks
|
|
rev: v4.6.0
|
|
hooks:
|
|
- id: check-yaml
|
|
- id: check-merge-conflict
|
|
- id: check-case-conflict
|
|
- id: trailing-whitespace
|
|
- id: end-of-file-fixer
|
|
- id: mixed-line-ending
|
|
args: ['--fix=lf']
|