"""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(		__UpperCAmelCase=None			):






	"""simple docstring"""





	if subparsers is not None:
		_lowercase :    Tuple							=							subparsers.add_parser('env'			)
	else:
		_lowercase :    Any							=							argparse.ArgumentParser('Accelerate env command'			)

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

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


def      __lowerCAmelCase(		__UpperCAmelCase			):






	"""simple docstring"""





	_lowercase :    Tuple							=							torch.__version__
	_lowercase :    List[Any]							=							torch.cuda.is_available()
	_lowercase :    str							=							is_xpu_available()
	_lowercase :    Dict							=							is_npu_available()

	_lowercase :    Union[str, Any]							=							'Not found'
	# Get the default from the config file.
	if args.config_file is not None or os.path.isfile(__UpperCAmelCase			):
		_lowercase :    List[Any]							=							load_config_from_file(args.config_file			).to_dict()

	_lowercase :    str							=							{
	    '`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(__UpperCAmelCase			),
	    'PyTorch NPU available': str(__UpperCAmelCase			),
	    'System RAM': F'''{psutil.virtual_memory().total / 1_024 ** 3:.2f} GB''',
	}
	if pt_cuda_available:
		_lowercase :    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:'			)
	_lowercase :    Optional[int]							=							(
	    '\n'.join([F'''\t- {prop}: {val}''' for prop, val in accelerate_config.items()]			)
	    if isinstance(__UpperCAmelCase     ,__UpperCAmelCase			)
	    else F'''\t{accelerate_config}'''
	)
	print(__UpperCAmelCase			)

	_lowercase :    Optional[Any]							=							accelerate_config

	return info


def      __lowerCAmelCase(		):






	"""simple docstring"""





	_lowercase :    str							=							env_command_parser()
	_lowercase :    Any							=							parser.parse_args()
	env_command(__UpperCAmelCase			)
	return 0


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




"""simple docstring"""







def      __lowerCAmelCase(		__UpperCAmelCase			):






	"""simple docstring"""





	_lowercase :    str							=							[0 for i in range(len(__UpperCAmelCase			)			)]

	# initialize interval's left pointer and right pointer
	_lowercase				,      _lowercase :    str							=							0, 0

	for i in range(1     ,len(__UpperCAmelCase			)			):
		# case when current index is inside the interval
		if i <= right_pointer:
			_lowercase :    Union[str, Any]							=							min(right_pointer - i + 1     ,z_result[i - left_pointer]			)
			_lowercase :    Any							=							min_edge

		while go_next(__UpperCAmelCase     ,__UpperCAmelCase     ,__UpperCAmelCase			):
			z_result[i] += 1

		# if new index's result gives us more right interval,
		# we've to update left_pointer and right_pointer
		if i + z_result[i] - 1 > right_pointer:
			_lowercase				,      _lowercase :    Optional[int]							=							i, i + z_result[i] - 1

	return z_result


def      __lowerCAmelCase(		__UpperCAmelCase     ,__UpperCAmelCase     ,__UpperCAmelCase			):






	"""simple docstring"""





	return i + z_result[i] < len(__UpperCAmelCase			) and s[z_result[i]] == s[i + z_result[i]]


def      __lowerCAmelCase(		__UpperCAmelCase     ,__UpperCAmelCase			):






	"""simple docstring"""





	_lowercase :    Union[str, Any]							=							0
	# concatenate 'pattern' and 'input_str' and call z_function
	# with concatenated string
	_lowercase :    List[str]							=							z_function(pattern + input_str			)

	for val in z_result:
		# if value is greater then length of the pattern string
		# that means this index is starting position of substring
		# which is equal to pattern string
		if val >= len(__UpperCAmelCase			):
			answer += 1

	return answer


if __name__ == "__main__":
						import doctest

						doctest.testmod()



