hxwang 803878b2fd [moe] full test for deepseek and mixtral (pp + sp to fix)

2024-08-01 10:06:59 +08: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

[pre-commit.ci] pre-commit autoupdate (#5572)

2024-07-01 17:16:41 +08:00

engine

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

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

inference

[pre-commit.ci] pre-commit autoupdate (#5572)

2024-07-01 17:16:41 +08:00

moe

[moe] full test for deepseek and mixtral (pp + sp to fix)

2024-08-01 10:06:59 +08:00

nn

[Hoxfix] Fix CUDA_DEVICE_MAX_CONNECTIONS for comm overlap

2024-07-05 20:02:36 +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