Mirrors/modelscope
1
0
Fork 0
mirror of https://github.com/modelscope/modelscope.git synced 2025-12-16 16:27:45 +01:00
Code Issues Packages Projects Releases Wiki Activity

remove profiling log (#113)

Browse Source 
UT test by default does not run profiling
This commit is contained in:
wenmeng zhou
2023-02-17 12:01:54 +08:00
committed by GitHub
parent 5df2930017
commit 3ca68bf22f
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
tests/run.py
View File

@@ -600,7 +600,7 @@ if __name__ == '__main__':
parser.add_argument(
'--level', default=0, type=int, help='2 -- all, 1 -- p1, 0 -- p0')
parser.add_argument(
'--disable_profile', action='store_true', help='disable profiling')
'--profile', action='store_true', help='enable profiling')
parser.add_argument(
'--run_config',
default=None,
@@ -634,7 +634,7 @@ if __name__ == '__main__':
set_test_level(args.level)
os.environ['REGRESSION_BASELINE'] = '1'
logger.info(f'TEST LEVEL: {test_level()}')
if not args.disable_profile:
if args.profile:
from utils import profiler
logger.info('enable profile ...')
profiler.enable()