


'''simple docstring'''

# Copyright 2022 The HuggingFace Team. All rights reserved.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

import argparse
import os
import platform

import numpy as np
import psutil
import torch

from accelerate import __version__ as version
from accelerate.commands.config import default_config_file, load_config_from_file

from ..utils import is_npu_available, is_xpu_available





def lowerCamelCase      (      lowerCAmelCase    :     Optional[int]=None		):





	"""simple docstring"""







	if subparsers is not None:
		__magic_name__	:   Tuple					     =					subparsers.add_parser('env'		)
	else:
		__magic_name__	:   Union[str, Any]					     =					argparse.ArgumentParser('Accelerate env command'		)

	parser.add_argument(
	    '--config_file'		,		default=lowerCAmelCase		,		help='The config file to use for the default values in the launching script.'		)

	if subparsers is not None:
		parser.set_defaults(func=lowerCAmelCase		)
	return parser





def lowerCamelCase      (      lowerCAmelCase    :     List[Any]		):





	"""simple docstring"""







	__magic_name__	:   Union[str, Any]					     =					torch.__version__
	__magic_name__	:   str					     =					torch.cuda.is_available()
	__magic_name__	:   Dict					     =					is_xpu_available()
	__magic_name__	:   Optional[Any]					     =					is_npu_available()

	__magic_name__	:   List[Any]					     =					'Not found'
	# Get the default from the config file.
	if args.config_file is not None or os.path.isfile(lowerCAmelCase		):
		__magic_name__	:   Dict					     =					load_config_from_file(args.config_file		).to_dict()

	__magic_name__	:   List[Any]					     =					{
	    '`Accelerate` version': version,
	    'Platform': platform.platform(),
	    'Python version': platform.python_version(),
	    'Numpy version': np.__version__,
	    'PyTorch version (GPU?)': f'{pt_version} ({pt_cuda_available})',
	    'PyTorch XPU available': str(lowerCAmelCase		),
	    'PyTorch NPU available': str(lowerCAmelCase		),
	    'System RAM': f'{psutil.virtual_memory().total / 1024 ** 3:.2f} GB',
	}
	if pt_cuda_available:
		__magic_name__	:   List[Any]					     =					torch.cuda.get_device_name()

	print('\nCopy-and-paste the text below in your GitHub issue\n'		)
	print('\n'.join([f'- {prop}: {val}' for prop, val in info.items()]		)		)

	print('- `Accelerate` default config:' if args.config_file is None else '- `Accelerate` config passed:'		)
	__magic_name__	:   Any					     =					(
	    '\n'.join([f'\t- {prop}: {val}' for prop, val in accelerate_config.items()]		)
	    if isinstance(lowerCAmelCase		,		lowerCAmelCase		)
	    else f'\t{accelerate_config}'
	)
	print(lowerCAmelCase		)

	__magic_name__	:   str					     =					accelerate_config

	return info





def lowerCamelCase      (      ):





	"""simple docstring"""







	__magic_name__	:   Optional[int]					     =					env_command_parser()
	__magic_name__	:   Tuple					     =					parser.parse_args()
	env_command(lowerCAmelCase		)
	return 0


if __name__ == "__main__":
							raise SystemExit(main())



'''simple docstring'''

# tests directory-specific settings - this file is run automatically
# by pytest before any tests are run

import doctest
import sys
import warnings
from os.path import abspath, dirname, join

import _pytest

from transformers.testing_utils import HfDoctestModule, HfDocTestParser


# allow having multiple repository checkouts and not needing to remember to rerun
# 'pip install -e .[dev]' when switching between checkouts and running tests.
lowerCAmelCase				:Optional[int]        =      abspath(join(dirname(__file__), '''src'''))
sys.path.insert(1, git_repo_path)

# silence FutureWarning warnings in tests since often we can't act on them until
# they become normal warnings - i.e. the tests still need to test the current functionality
warnings.simplefilter(action='''ignore''', category=FutureWarning)





def lowerCamelCase      (      lowerCAmelCase    :     Tuple		):





	"""simple docstring"""







	config.addinivalue_line(
	    'markers'		,		'is_pt_tf_cross_test: mark test to run only when PT and TF interactions are tested'		)
	config.addinivalue_line(
	    'markers'		,		'is_pt_flax_cross_test: mark test to run only when PT and FLAX interactions are tested'		)
	config.addinivalue_line('markers'		,		'is_pipeline_test: mark test to run only when pipelines are tested'		)
	config.addinivalue_line('markers'		,		'is_staging_test: mark test to run only in the staging environment'		)
	config.addinivalue_line('markers'		,		'accelerate_tests: mark test that require accelerate'		)
	config.addinivalue_line('markers'		,		'tool_tests: mark the tool tests that are run on their specific schedule'		)





def lowerCamelCase      (      lowerCAmelCase    :     Any		):





	"""simple docstring"""







	from transformers.testing_utils import pytest_addoption_shared

	pytest_addoption_shared(lowerCAmelCase		)





def lowerCamelCase      (      lowerCAmelCase    :     Dict		):





	"""simple docstring"""







	from transformers.testing_utils import pytest_terminal_summary_main

	__magic_name__	:   Tuple					     =					terminalreporter.config.getoption('--make-reports'		)
	if make_reports:
		pytest_terminal_summary_main(lowerCAmelCase		,		id=lowerCAmelCase		)





def lowerCamelCase      (      lowerCAmelCase    :     List[Any]		,		lowerCAmelCase    :     Tuple		):





	"""simple docstring"""







	if exitstatus == 5:
		__magic_name__	:   Any					     =					0


# Doctest custom flag to ignore output.
lowerCAmelCase				:List[str]        =      doctest.register_optionflag('''IGNORE_RESULT''')

lowerCAmelCase				:Union[str, Any]        =      doctest.OutputChecker
class _lowerCamelCase	(						lowercase__							):



		'''simple docstring'''







		def        __lowerCAmelCase		( self		:						List[str]  ,					_A		:						Tuple  ,					_A		:						Tuple  ,					_A		:						str      )  ->			int:
			if IGNORE_RESULT & optionflags:
				return True
			return OutputChecker.check_output(self  ,					_A  ,					_A  ,					_A      )







lowerCAmelCase				:Optional[Any]        =      CustomOutputChecker
lowerCAmelCase				:int        =      HfDoctestModule





lowerCAmelCase				:Any        =      HfDocTestParser