You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
ColossalAI/colossalai
Hongxin Liu 3e05c07bb8
[lazy] support torch 2.0 (#4763)
1 year ago
..
_C [setup] support pre-build and jit-build of cuda kernels (#2374) 2 years ago
_analyzer [misc] update pre-commit and run all files (#4752) 1 year ago
amp [shardformer] fix master param sync for hybrid plugin/rewrite unwrapping logic (#4758) 1 year ago
auto_parallel [misc] update pre-commit and run all files (#4752) 1 year ago
autochunk [misc] update pre-commit and run all files (#4752) 1 year ago
booster [shardformer] fix master param sync for hybrid plugin/rewrite unwrapping logic (#4758) 1 year ago
checkpoint_io [shardformer] fix master param sync for hybrid plugin/rewrite unwrapping logic (#4758) 1 year ago
cli [misc] update pre-commit and run all files (#4752) 1 year ago
cluster [misc] update pre-commit and run all files (#4752) 1 year ago
context [misc] update pre-commit and run all files (#4752) 1 year ago
device [misc] update pre-commit and run all files (#4752) 1 year ago
fx [misc] update pre-commit and run all files (#4752) 1 year ago
inference [misc] update pre-commit and run all files (#4752) 1 year ago
interface [misc] update pre-commit and run all files (#4752) 1 year ago
kernel [misc] update pre-commit and run all files (#4752) 1 year ago
lazy [lazy] support torch 2.0 (#4763) 1 year ago
legacy [bug] fix get_default_parser in examples (#4764) 1 year ago
logging [misc] update pre-commit and run all files (#4752) 1 year ago
nn [misc] update pre-commit and run all files (#4752) 1 year ago
pipeline [misc] update pre-commit and run all files (#4752) 1 year ago
shardformer [misc] update pre-commit and run all files (#4752) 1 year ago
tensor [misc] update pre-commit and run all files (#4752) 1 year ago
testing [misc] update pre-commit and run all files (#4752) 1 year ago
utils [misc] update pre-commit and run all files (#4752) 1 year ago
zero [shardformer] fix master param sync for hybrid plugin/rewrite unwrapping logic (#4758) 1 year ago
__init__.py [misc] update pre-commit and run all files (#4752) 1 year ago
initialize.py [misc] update pre-commit and run all files (#4752) 1 year ago