mirror of
https://github.com/hpcaitech/ColossalAI.git
synced 2025-09-01 17:17:05 +00:00
[profiler] primary memory tracer
This commit is contained in:
@@ -22,6 +22,7 @@ def test_load_config():
|
||||
|
||||
@pytest.mark.cpu
|
||||
def test_load_ophooks():
|
||||
dict = {'type': 'MemTracerOpHook', 'niter': 2}
|
||||
dict = {'type': 'MemTracerOpHook', 'warmup': 10, 'refreshrate': 20}
|
||||
ophook = build_ophooks(dict)
|
||||
assert ophook.niter() == 2
|
||||
assert ophook.refreshrate == 20
|
||||
assert ophook.warmup == 10
|
||||
|
Reference in New Issue
Block a user