/
githubmirror
/
ColossalAI
Обзор
Документация
Войти
/
githubmirror
/
ColossalAI
Код
Запросы
0
Пакеты
0
Релизы
0
Аналитика
Безопасность
v0.5.1
tests/conftest.py
10 строк
231 B
Hongxin Liu
[checkpointio] support load-pin overlap (#6177)
07 янв 2025, 11:16
Не верифицирован
07 янв 2025, 11:16
ee81366
Код
Авторство
О чём код?
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()