mirror of https://github.com/hpcaitech/ColossalAI
11 lines
231 B
Python
11 lines
231 B
Python
|
import gc
|
||
|
|
|
||
|
|
from colossalai.accelerator import get_accelerator
|
||
|
|
|
||
|
|
|
||
|
|
def pytest_runtest_setup(item):
|
||
|
|
# called for running each test in 'a' directory
|
||
|
|
accelerator = get_accelerator()
|
||
|
|
accelerator.empty_cache()
|
||
|
|
gc.collect()
|