flags.py 948 B

12345678910111213141516171819202122232425262728293031
  1. # !/usr/bin/env python3
  2. # -*- coding: UTF-8 -*-
  3. ################################################################################
  4. #
  5. # Copyright (c) 2024 Baidu.com, Inc. All Rights Reserved
  6. #
  7. ################################################################################
  8. """
  9. Author: PaddlePaddle Authors
  10. """
  11. import os
  12. __all__ = ['DEBUG', 'DRY_RUN', 'CHECK_OPTS', 'EAGER_INITIALIZATION']
  13. def get_flag_from_env_var(name, default):
  14. """ get_flag_from_env_var """
  15. env_var = os.environ.get(name, None)
  16. if env_var in ('True', 'true', 'TRUE', '1'):
  17. return True
  18. elif env_var in ('False', 'false', 'FALSE', '0'):
  19. return False
  20. else:
  21. return default
  22. DEBUG = get_flag_from_env_var('PADDLE_PDX_DEBUG', False)
  23. DRY_RUN = get_flag_from_env_var('PADDLE_PDX_DRY_RUN', False)
  24. CHECK_OPTS = get_flag_from_env_var('PADDLE_PDX_CHECK_OPTS', False)
  25. EAGER_INITIALIZATION = get_flag_from_env_var('PADDLE_PDX_EAGER_INIT', True)