github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-09-06 03:20:52 +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
2 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
tests/test_device/test_device_mesh.py
View File

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