Edenzzzz 61da3fbc52 fixed layout converter caching and updated tester

2024-03-26 17:22:27 +08:00

..

d_tensor

fixed layout converter caching and updated tester

2024-03-26 17:22:27 +08:00

moe_tensor

[moe] support mixtral (#5309)

2024-02-07 19:21:02 +08:00

__init__.py

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

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

colo_parameter.py

[gemini] fix param op hook when output is tuple (#5355)

2024-02-04 11:58:26 +08:00

colo_tensor.py

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

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

comm_spec.py

fix some typo (#5307)

2024-01-25 13:56:27 +08:00

param_op_hook.py

[gemini] fix param op hook when output is tuple (#5355)

2024-02-04 11:58:26 +08:00

shape_consistency.py

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

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

sharding_spec.py

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

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

utils.py

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

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