Guangyao Zhang 2e28c793ce [compatibility] support torch 2.2 (#5875) ...

* Support Pytorch 2.2.2

* keep build_on_pr file and update .compatibility

2024-07-16 13:59:25 +08:00

..

d_tensor

[compatibility] support torch 2.2 (#5875)

2024-07-16 13:59:25 +08:00

moe_tensor

[MoE/ZeRO] Moe refactor with zero refactor (#5821)

2024-06-28 14:00:08 +08:00

padded_tensor

[shardformer] refactor embedding resize (#5603)

2024-04-18 16:10:18 +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

[Auto Parallel]: Speed up intra-op plan generation by 44% (#5446)

2024-07-15 12:05:06 +08:00

utils.py

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

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