



'''simple docstring'''


import torch

from accelerate import PartialState
from accelerate.utils.operations import broadcast, gather, gather_object, pad_across_processes, reduce







def      SCREAMING_SNAKE_CASE				(				lowercase_    :      Optional[Any]						):
						return (torch.arange(state.num_processes						) + 1.0 + (state.num_processes * state.process_index)).to(state.device						)







def      SCREAMING_SNAKE_CASE				(				lowercase_    :      Union[str, Any]						):
						lowercase        =							create_tensor(_snake_case						)
						lowercase        =							gather(_snake_case						)
						assert gathered_tensor.tolist() == list(range(1 ,  state.num_processes**2 + 1						)						)







def      SCREAMING_SNAKE_CASE				(				lowercase_    :      Union[str, Any]						):
						lowercase        =							[state.process_index]
						lowercase        =							gather_object(_snake_case						)
						assert len(_snake_case						) == state.num_processes, F"""{gathered_obj}, {len(_snake_case						)} != {state.num_processes}"""
						assert gathered_obj == list(range(state.num_processes						)						), F"""{gathered_obj} != {list(range(state.num_processes						)						)}"""







def      SCREAMING_SNAKE_CASE				(				lowercase_    :      Tuple						):
						lowercase        =							create_tensor(_snake_case						)
						lowercase        =							broadcast(_snake_case						)
						assert broadcasted_tensor.shape == torch.Size([state.num_processes]						)
						assert broadcasted_tensor.tolist() == list(range(1 ,  state.num_processes + 1						)						)







def      SCREAMING_SNAKE_CASE				(				lowercase_    :      Optional[int]						):






						if state.is_main_process:
												lowercase        =							torch.arange(state.num_processes + 1						).to(state.device						)
						else:
												lowercase        =							torch.arange(state.num_processes						).to(state.device						)
						lowercase        =							pad_across_processes(_snake_case						)
						assert padded_tensor.shape == torch.Size([state.num_processes + 1]						)
						if not state.is_main_process:
												assert padded_tensor.tolist() == list(range(0 ,  state.num_processes						)						) + [0]







def      SCREAMING_SNAKE_CASE				(				lowercase_    :      Optional[Any]						):






						if state.num_processes != 2:
												return
						lowercase        =							create_tensor(_snake_case						)
						lowercase        =							reduce(_snake_case ,  """sum"""						)
						lowercase        =							torch.tensor([4.0, 6]						).to(state.device						)
						assert torch.allclose(_snake_case ,  _snake_case						), F"""{reduced_tensor} != {truth_tensor}"""







def      SCREAMING_SNAKE_CASE				(				lowercase_    :      Optional[int]						):






						if state.num_processes != 2:
												return
						lowercase        =							create_tensor(_snake_case						)
						lowercase        =							reduce(_snake_case ,  """mean"""						)
						lowercase        =							torch.tensor([2.0, 3]						).to(state.device						)
						assert torch.allclose(_snake_case ,  _snake_case						), F"""{reduced_tensor} != {truth_tensor}"""







def      SCREAMING_SNAKE_CASE				(				lowercase_    :      Optional[Any]						):
						main()







def      SCREAMING_SNAKE_CASE				(				):
						lowercase        =							PartialState()
						state.print(F"""State: {state}"""						)
						state.print("""testing gather"""						)
						test_gather(_snake_case						)
						state.print("""testing gather_object"""						)
						test_gather_object(_snake_case						)
						state.print("""testing broadcast"""						)
						test_broadcast(_snake_case						)
						state.print("""testing pad_across_processes"""						)
						test_pad_across_processes(_snake_case						)
						state.print("""testing reduce_sum"""						)
						test_reduce_sum(_snake_case						)
						state.print("""testing reduce_mean"""						)
						test_reduce_mean(_snake_case						)


if __name__ == "__main__":
			main()







'''simple docstring'''


def      SCREAMING_SNAKE_CASE				(				lowercase_    :      Union[str, Any] ,  lowercase_    :      str						):
						lowercase        =							""""""
						for i in table:
												res += inp[i - 1]
						return res







def      SCREAMING_SNAKE_CASE				(				lowercase_    :      List[Any]						):
						return data[1:] + data[0]







def      SCREAMING_SNAKE_CASE				(				lowercase_    :      Tuple ,  lowercase_    :      Dict						):
						lowercase        =							""""""
						for i in range(len(lowercase_						)						):
												if a[i] == b[i]:
																		res += "0"
												else:
																		res += "1"
						return res







def      SCREAMING_SNAKE_CASE				(				lowercase_    :      str ,  lowercase_    :      str						):
						lowercase        =							int("""0b""" + data[0] + data[-1] ,  2						)
						lowercase        =							int("""0b""" + data[1:3] ,  2						)
						return bin(s[row][col]						)[2:]







def      SCREAMING_SNAKE_CASE				(				lowercase_    :      Dict ,  lowercase_    :      Tuple ,  lowercase_    :      Tuple ,  lowercase_    :      str ,  lowercase_    :      Any						):
						lowercase        =							message[:4]
						lowercase        =							message[4:]
						lowercase        =							apply_table(lowercase_ ,  lowercase_						)
						lowercase        =							xor(lowercase_ ,  lowercase_						)
						lowercase        =							apply_sbox(lowercase_ ,  temp[:4]						)  # noqa: E741
						lowercase        =							apply_sbox(lowercase_ ,  temp[4:]						)
						lowercase        =							"""0""" * (2 - len(lowercase_						)) + l  # noqa: E741
						lowercase        =							"""0""" * (2 - len(lowercase_						)) + r
						lowercase        =							apply_table(l + r ,  lowercase_						)
						lowercase        =							xor(lowercase_ ,  lowercase_						)
						return temp + right


if __name__ == "__main__":
						lowercase_      :				Tuple   					=    input('''Enter 10 bit key: ''')
						lowercase_      :				Any   					=    input('''Enter 8 bit message: ''')

						lowercase_      :				Dict   					=    [6, 3, 7, 4, 8, 5, 10, 9]
						lowercase_      :				str   					=    [3, 5, 2, 7, 4, 10, 1, 9, 8, 6]
						lowercase_      :				List[Any]   					=    [2, 4, 3, 1]
						lowercase_      :				List[str]   					=    [2, 6, 3, 1, 4, 8, 5, 7]
						lowercase_      :				Tuple   					=    [4, 1, 3, 5, 7, 2, 8, 6]
						lowercase_      :				Optional[Any]   					=    [4, 1, 2, 3, 2, 3, 4, 1]
						lowercase_      :				List[str]   					=    [[1, 0, 3, 2], [3, 2, 1, 0], [0, 2, 1, 3], [3, 1, 3, 2]]
						lowercase_      :				List[Any]   					=    [[0, 1, 2, 3], [2, 0, 1, 3], [3, 0, 1, 0], [2, 1, 0, 3]]

						# key generation
						lowercase_      :				Union[str, Any]   					=    apply_table(key, paa_table)
						lowercase_      :				Optional[Any]   					=    temp[:5]
						lowercase_      :				int   					=    temp[5:]
						lowercase_      :				List[str]   					=    left_shift(left)
						lowercase_      :				int   					=    left_shift(right)
						lowercase_      :				Tuple   					=    apply_table(left + right, pa_table)
						lowercase_      :				List[str]   					=    left_shift(left)
						lowercase_      :				Optional[Any]   					=    left_shift(right)
						lowercase_      :				Union[str, Any]   					=    left_shift(left)
						lowercase_      :				Union[str, Any]   					=    left_shift(right)
						lowercase_      :				Optional[int]   					=    apply_table(left + right, pa_table)

						# encryption
						lowercase_      :				int   					=    apply_table(message, IP)
						lowercase_      :				Dict   					=    function(expansion, sa, sa, keya, temp)
						lowercase_      :				Any   					=    temp[4:] + temp[:4]
						lowercase_      :				List[Any]   					=    function(expansion, sa, sa, keya, temp)
						lowercase_      :				Tuple   					=    apply_table(temp, IP_inv)
						print('''Cipher text is:''', CT)

						# decryption
						lowercase_      :				List[str]   					=    apply_table(CT, IP)
						lowercase_      :				Optional[int]   					=    function(expansion, sa, sa, keya, temp)
						lowercase_      :				Optional[Any]   					=    temp[4:] + temp[:4]
						lowercase_      :				Optional[int]   					=    function(expansion, sa, sa, keya, temp)
						lowercase_      :				Optional[Any]   					=    apply_table(temp, IP_inv)
						print('''Plain text after decypting is:''', PT)


