github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-06-28 16:28:10 +00:00
Code Issues Packages Projects Releases Wiki Activity

[format] applied code formatting on changed files in pull request 4021 (#4022)

Browse Source 
Co-authored-by: github-actions <github-actions@github.com>
This commit is contained in:
github-actions[bot] 2023-06-19 11:23:24 +08:00 committed by GitHub
parent 160c64c645
commit a52f62082d
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
colossalai/nn/layer/base_layer.py
View File

@ -1,11 +1,12 @@
#!/usr/bin/env python #!/usr/bin/env python
# -*- encoding: utf-8 -*- # -*- encoding: utf-8 -*-
from contextlib import contextmanager
import torch.nn as nn import torch.nn as nn
from colossalai.context import ParallelMode from colossalai.context import ParallelMode
from colossalai.core import global_context as gpc from colossalai.core import global_context as gpc
from contextlib import contextmanager
class ParallelLayer(nn.Module): class ParallelLayer(nn.Module):

3
tests/test_device/test_device_mesh.py
View File

@ -1,6 +1,7 @@
from colossalai.device.device_mesh import DeviceMesh
import torch import torch
from colossalai.device.device_mesh import DeviceMesh
def test_device_mesh(): def test_device_mesh():
physical_mesh_id = torch.arange(0, 16).reshape(2, 8) physical_mesh_id = torch.arange(0, 16).reshape(2, 8)