mirror of
https://github.com/hpcaitech/ColossalAI.git
synced 2025-09-15 14:12:02 +00:00
[misc] update pre-commit and run all files (#4752)
* [misc] update pre-commit * [misc] run pre-commit * [misc] remove useless configuration files * [misc] ignore cuda for clang-format
This commit is contained in:
@@ -38,7 +38,9 @@ def check_equal(A, B):
|
||||
|
||||
|
||||
def check_checkpoint_1d(rank, world_size, port):
|
||||
config = dict(parallel=dict(pipeline=dict(size=2), tensor=dict(size=4, mode="1d")),)
|
||||
config = dict(
|
||||
parallel=dict(pipeline=dict(size=2), tensor=dict(size=4, mode="1d")),
|
||||
)
|
||||
|
||||
disable_existing_loggers()
|
||||
launch(config=config, rank=rank, world_size=world_size, host="localhost", port=port, backend="nccl")
|
||||
|
@@ -38,7 +38,9 @@ def check_equal(A, B):
|
||||
|
||||
|
||||
def check_checkpoint_2d(rank, world_size, port):
|
||||
config = dict(parallel=dict(pipeline=dict(size=2), tensor=dict(size=4, mode="2d")),)
|
||||
config = dict(
|
||||
parallel=dict(pipeline=dict(size=2), tensor=dict(size=4, mode="2d")),
|
||||
)
|
||||
|
||||
disable_existing_loggers()
|
||||
launch(config=config, rank=rank, world_size=world_size, host="localhost", port=port, backend="nccl")
|
||||
|
@@ -38,7 +38,9 @@ def check_equal(A, B):
|
||||
|
||||
|
||||
def check_checkpoint_2p5d(rank, world_size, port):
|
||||
config = dict(parallel=dict(pipeline=dict(size=2), tensor=dict(size=4, depth=1, mode="2.5d")),)
|
||||
config = dict(
|
||||
parallel=dict(pipeline=dict(size=2), tensor=dict(size=4, depth=1, mode="2.5d")),
|
||||
)
|
||||
|
||||
disable_existing_loggers()
|
||||
launch(config=config, rank=rank, world_size=world_size, host="localhost", port=port, backend="nccl")
|
||||
|
@@ -38,7 +38,9 @@ def check_equal(A, B):
|
||||
|
||||
|
||||
def check_checkpoint_3d(rank, world_size, port):
|
||||
config = dict(parallel=dict(pipeline=dict(size=1), tensor=dict(size=8, mode="3d")),)
|
||||
config = dict(
|
||||
parallel=dict(pipeline=dict(size=1), tensor=dict(size=8, mode="3d")),
|
||||
)
|
||||
|
||||
disable_existing_loggers()
|
||||
launch(config=config, rank=rank, world_size=world_size, host="localhost", port=port, backend="nccl")
|
||||
|
Reference in New Issue
Block a user