/
githubmirror
/
ColossalAI
Обзор
Документация
Войти
/
githubmirror
/
ColossalAI
Код
Запросы
0
Пакеты
0
Релизы
0
Аналитика
Безопасность
main
colossalai/accelerator/__init__.py
15 строк
431 B
Hongxin Liu
[npu] change device to accelerator api (#5239)
09 янв 2024, 05:20
Не верифицирован
09 янв 2024, 05:20
d202cc2
Код
Авторство
О чём код?
from .api import auto_set_accelerator, get_accelerator, set_accelerator from .base_accelerator import BaseAccelerator from .cpu_accelerator import CpuAccelerator from .cuda_accelerator import CudaAccelerator from .npu_accelerator import NpuAccelerator __all__ = [ "get_accelerator", "set_accelerator", "auto_set_accelerator", "BaseAccelerator", "CudaAccelerator", "NpuAccelerator", "CpuAccelerator", ]