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/legacy/nn
Edenzzzz 8ec24b6a4d
[Hoxfix] Fix CUDA_DEVICE_MAX_CONNECTIONS for comm overlap
5 months ago
..
_ops [misc] update pre-commit and run all files (#4752) 1 year ago
layer [Hoxfix] Fix CUDA_DEVICE_MAX_CONNECTIONS for comm overlap 5 months ago
loss [npu] change device to accelerator api (#5239) 11 months ago
metric [misc] update pre-commit and run all files (#4752) 1 year ago
parallel [misc] update pre-commit and run all files (#4752) 1 year ago
__init__.py [legacy] clean up legacy code (#4743) 1 year ago