github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-09-08 04:24:47 +00:00
Code Issues Packages Projects Releases Wiki Activity

added testing module (#435)

Browse Source
This commit is contained in:
Frank Lee
2022-03-16 17:20:05 +08:00
committed by GitHub
parent dbdc9a7783
commit bffd85bf34
3 changed files with 85 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
colossalai/testing/__init__.py Normal file
View File

@@ -0,0 +1,4 @@
from .comparison import assert_equal, assert_not_equal, assert_close, assert_close_loose, assert_equal_in_group
from .utils import parameterize
__all__ = ['assert_equal', 'assert_not_equal', 'assert_close', 'assert_close_loose', 'assert_equal_in_group', 'parameterize']