github
/
ColossalAI
mirror of https://github.com/hpcaitech/ColossalAI
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

[NFC] polish tests/test_layers/test_3d/checks_3d/common.py code style (#1733)

pull/1743/head
lucasliunju 2022-10-18 21:49:41 +08:00 committed by Frank Lee
parent 23703c9dd6
commit 912eb58ea0
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
tests/test_layers/test_3d/checks_3d/common.py
View File

@ -16,3 +16,4 @@ def check_equal(A, B):
eq = torch.allclose(A, B, rtol=1e-3, atol=1e-2) eq = torch.allclose(A, B, rtol=1e-3, atol=1e-2)
assert eq assert eq
return eq return eq