__init__.py 298 B

12345678910111213141516171819
  1. # Copyright (c) Opendatalab. All rights reserved.
  2. """工具函数模块"""
  3. from .device_env import (
  4. DeviceKind,
  5. detect_device_kind,
  6. is_nvidia,
  7. is_npu,
  8. is_cpu_only,
  9. )
  10. __all__ = [
  11. "DeviceKind",
  12. "detect_device_kind",
  13. "is_nvidia",
  14. "is_npu",
  15. "is_cpu_only",
  16. ]