Yuanheng Zhao 8633c15da9 [sync] Sync feature/colossal-infer with main

2024-05-20 15:50:53 +00:00

..

amp

[feat] refactored extension module (#5298)

2024-01-25 17:01:48 +08:00

builder

[misc] update pre-commit and run all files (#4752)

2023-09-19 14:20:26 +08:00

communication

[npu] change device to accelerator api (#5239)

2024-01-09 10:20:05 +08:00

context

[hotfix] fix torch 2.0 compatibility (#4936)

2023-10-18 11:05:25 +08:00

engine

[npu] change device to accelerator api (#5239)

2024-01-09 10:20:05 +08:00

inference

[sync] Sync feature/colossal-infer with main

2024-05-20 15:50:53 +00:00

nn

[feat] refactored extension module (#5298)

2024-01-25 17:01:48 +08:00

pipeline

[misc] refactor launch API and tensor constructor (#5666)

2024-04-29 10:40:11 +08:00

registry

[misc] update pre-commit and run all files (#4752)

2023-09-19 14:20:26 +08:00

tensor

Fix ColoTensorSpec for py11 (#5440)

2024-03-26 15:56:49 +08:00

trainer

[npu] change device to accelerator api (#5239)

2024-01-09 10:20:05 +08:00

utils

[feat] refactored extension module (#5298)

2024-01-25 17:01:48 +08:00

zero

Merge branch 'main' into sync/npu

2024-01-18 12:05:21 +08:00

__init__.py

[bug] fix get_default_parser in examples (#4764)

2023-09-21 10:42:25 +08:00

constants.py

[misc] update pre-commit and run all files (#4752)

2023-09-19 14:20:26 +08:00

core.py

[misc] update pre-commit and run all files (#4752)

2023-09-19 14:20:26 +08:00

global_variables.py

[misc] update pre-commit and run all files (#4752)

2023-09-19 14:20:26 +08:00

initialize.py

[npu] change device to accelerator api (#5239)

2024-01-09 10:20:05 +08:00