


# 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					(SCREAMING_SNAKE_CASE=None      )->						Any:




      """simple docstring"""

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

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

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






def  __lowerCAmelCase					(SCREAMING_SNAKE_CASE      )->						List[Any]:




      """simple docstring"""

      snake_case_ =   torch.__version__
      snake_case_ =   torch.cuda.is_available()
      snake_case_ =   is_xpu_available()
      snake_case_ =   is_npu_available()

      snake_case_ =   '''Not found'''
      # Get the default from the config file.
      if args.config_file is not None or os.path.isfile(SCREAMING_SNAKE_CASE      ):
            snake_case_ =   load_config_from_file(args.config_file      ).to_dict()

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

      snake_case_ =   accelerate_config

      return info






def  __lowerCAmelCase					()->						int:




      """simple docstring"""

      snake_case_ =   env_command_parser()
      snake_case_ =   parser.parse_args()
      env_command(SCREAMING_SNAKE_CASE      )
      return 0


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



def  __lowerCAmelCase					(SCREAMING_SNAKE_CASE      ,      SCREAMING_SNAKE_CASE      )->						int:




      """simple docstring"""

      return int((input_a, input_a).count(0      ) == 0      )






def  __lowerCAmelCase					()->						None:




      """simple docstring"""

      assert and_gate(0      ,      0      ) == 0
      assert and_gate(0      ,      1      ) == 0
      assert and_gate(1      ,      0      ) == 0
      assert and_gate(1      ,      1      ) == 1


if __name__ == "__main__":
   test_and_gate()
   print(and_gate(1, 0))
   print(and_gate(0, 0))
   print(and_gate(0, 1))
   print(and_gate(1, 1))