




def 		lowercase__ (		__A:						list[int]					):





					'''simple docstring'''






					if len(__A					) > 1:
										__magic_name__				:						List[Any]     				=				len(__A					) // 2  # Finds the middle of the array
										__magic_name__				:						Union[str, Any]     				=				arr[
										    :middle_length
										]  # Creates an array of the elements in the first half.
										__magic_name__				:						Dict     				=				arr[
										    middle_length:
										]  # Creates an array of the elements in the second half.
										__magic_name__				:						List[Any]     				=				len(__A					)
										__magic_name__				:						Union[str, Any]     				=				len(__A					)
										merge(__A					)  # Starts sorting the left.
										merge(__A					)  # Starts sorting the right
										__magic_name__				:						Optional[Any]     				=				0  # Left Counter
										__magic_name__				:						int     				=				0  # Right Counter
										__magic_name__				:						Tuple     				=				0  # Position Counter
										while (
										    left_index < left_size and right_index < right_size
										):  # Runs until the lowers size of the left and right are sorted.
															if left_array[left_index] < right_array[right_index]:
																				__magic_name__				:						str     				=				left_array[left_index]
																				left_index += 1
															else:
																				__magic_name__				:						List[Any]     				=				right_array[right_index]
																				right_index += 1
															index += 1
										while (
										    left_index < left_size
										):  # Adds the left over elements in the left half of the array
															__magic_name__				:						str     				=				left_array[left_index]
															left_index += 1
															index += 1
										while (
										    right_index < right_size
										):  # Adds the left over elements in the right half of the array
															__magic_name__				:						str     				=				right_array[right_index]
															right_index += 1
															index += 1
					return arr


if __name__ == "__main__":
	import doctest

	doctest.testmod()









import tempfile
import unittest

import transformers
from transformers import XGLMConfig, XGLMTokenizer, is_flax_available, is_torch_available
from transformers.testing_utils import is_pt_flax_cross_test, require_flax, require_sentencepiece, slow

from ...generation.test_flax_utils import FlaxGenerationTesterMixin
from ...test_modeling_flax_common import FlaxModelTesterMixin, floats_tensor, ids_tensor, random_attention_mask


if is_flax_available():
	import jax
	import jax.numpy as jnp
	import numpy as np

	from transformers.modeling_flax_pytorch_utils import (
	    convert_pytorch_state_dict_to_flax,
	    load_flax_weights_in_pytorch_model,
	)
	from transformers.models.xglm.modeling_flax_xglm import FlaxXGLMForCausalLM, FlaxXGLMModel


if is_torch_available():
	import torch


@require_flax
class  lowerCamelCase     :






			'''simple docstring'''





			def __init__(			self      : Union[str, Any]							,    lowerCamelCase_      : List[Any]							,    lowerCamelCase_      : str=14							,    lowerCamelCase_      : str=7							,    lowerCamelCase_      : Dict=True							,    lowerCamelCase_      : Union[str, Any]=True							,    lowerCamelCase_      : Optional[int]=True							,    lowerCamelCase_      : Optional[int]=99							,    lowerCamelCase_      : List[str]=32							,    lowerCamelCase_      : Tuple=5							,    lowerCamelCase_      : str=4							,    lowerCamelCase_      : Optional[Any]=37							,    lowerCamelCase_      : str="gelu"							,    lowerCamelCase_      : List[str]=0.1							,    lowerCamelCase_      : Optional[Any]=0.1							,    lowerCamelCase_      : Any=512							,    lowerCamelCase_      : Optional[int]=0.0_2							,    lowerCamelCase_      : Dict=None							,    )    -> Optional[int]:
								__magic_name__				:						List[Any]     				=				parent
								__magic_name__				:						int     				=				batch_size
								__magic_name__				:						Union[str, Any]     				=				seq_length
								__magic_name__				:						Dict     				=				is_training
								__magic_name__				:						List[str]     				=				use_input_mask
								__magic_name__				:						Optional[Any]     				=				use_labels
								__magic_name__				:						List[Any]     				=				vocab_size
								__magic_name__				:						Optional[int]     				=				d_model
								__magic_name__				:						Tuple     				=				num_hidden_layers
								__magic_name__				:						Any     				=				num_attention_heads
								__magic_name__				:						int     				=				ffn_dim
								__magic_name__				:						List[Any]     				=				activation_function
								__magic_name__				:						Dict     				=				activation_dropout
								__magic_name__				:						int     				=				attention_dropout
								__magic_name__				:						List[str]     				=				max_position_embeddings
								__magic_name__				:						Union[str, Any]     				=				initializer_range
								__magic_name__				:						Optional[Any]     				=				None
								__magic_name__				:						Tuple     				=				0
								__magic_name__				:						List[str]     				=				2
								__magic_name__				:						Dict     				=				1

			def 				UpperCAmelCase__							(			self      : int	)    -> int:
								__magic_name__				:						Tuple     				=				np.clip(ids_tensor([self.batch_size, self.seq_length]							,    self.vocab_size	)							,    3							,    self.vocab_size	)

								__magic_name__				:						Any     				=				None
								if self.use_input_mask:
													__magic_name__				:						Optional[int]     				=				random_attention_mask([self.batch_size, self.seq_length]	)

								__magic_name__				:						Union[str, Any]     				=				XGLMConfig(
								    vocab_size=self.vocab_size							,    d_model=self.hidden_size							,    num_layers=self.num_hidden_layers							,    attention_heads=self.num_attention_heads							,    ffn_dim=self.ffn_dim							,    activation_function=self.activation_function							,    activation_dropout=self.activation_dropout							,    attention_dropout=self.attention_dropout							,    max_position_embeddings=self.max_position_embeddings							,    initializer_range=self.initializer_range							,    use_cache=lowerCamelCase_							,    bos_token_id=self.bos_token_id							,    eos_token_id=self.eos_token_id							,    pad_token_id=self.pad_token_id							,    )

								return (config, input_ids, input_mask)

			def 				UpperCAmelCase__							(			self      : Union[str, Any]	)    -> Tuple:
								__magic_name__				:						int     				=				self.prepare_config_and_inputs()
								__magic_name__				,     __magic_name__				,     __magic_name__				:						Tuple     				=				config_and_inputs
								__magic_name__				:						Any     				=				{'''input_ids''': input_ids, '''attention_mask''': attention_mask}
								return config, inputs_dict

			def 				UpperCAmelCase__							(			self      : Dict	)    -> Dict:
								__magic_name__				,     __magic_name__				,     __magic_name__				:						Dict     				=				self.prepare_config_and_inputs()

								__magic_name__				:						Any     				=				floats_tensor([self.batch_size, self.seq_length, self.hidden_size]	)
								__magic_name__				:						Optional[Any]     				=				ids_tensor([self.batch_size, self.seq_length]							,    vocab_size=2	)

								return (
								    config,
								    input_ids,
								    attention_mask,
								    encoder_hidden_states,
								    encoder_attention_mask,
								)

			def 				UpperCAmelCase__							(			self      : str							,    lowerCamelCase_      : Tuple							,    lowerCamelCase_      : Optional[int]							,    lowerCamelCase_      : Optional[int]							,    lowerCamelCase_      : Any	)    -> Dict:
								__magic_name__				:						Dict     				=				20
								__magic_name__				:						List[str]     				=				model_class_name(lowerCamelCase_	)

								__magic_name__				:						Any     				=				model.init_cache(input_ids.shape[0]							,    lowerCamelCase_	)
								__magic_name__				:						Tuple     				=				jnp.ones((input_ids.shape[0], max_decoder_length)							,    dtype='''i4'''	)

								__magic_name__				:						Union[str, Any]     				=				jnp.broadcast_to(
								    jnp.arange(input_ids.shape[-1] - 1	)[None, :]							,    (input_ids.shape[0], input_ids.shape[-1] - 1)	)
								__magic_name__				:						Dict     				=				model(
								    input_ids[:, :-1]							,    attention_mask=lowerCamelCase_							,    past_key_values=lowerCamelCase_							,    position_ids=lowerCamelCase_							,    )

								__magic_name__				:						Dict     				=				jnp.array(input_ids.shape[0] * [[input_ids.shape[-1] - 1]]							,    dtype='''i4'''	)
								__magic_name__				:						int     				=				model(
								    input_ids[:, -1:]							,    attention_mask=lowerCamelCase_							,    past_key_values=outputs_cache.past_key_values							,    position_ids=lowerCamelCase_							,    )

								__magic_name__				:						Optional[int]     				=				model(lowerCamelCase_	)

								__magic_name__				:						Union[str, Any]     				=				np.max(np.abs((outputs_cache_next[0][:, -1, :5] - outputs[0][:, -1, :5])	)	)
								self.parent.assertTrue(diff < 1E-3							,    msg=F'''Max diff is {diff}'''	)


			def 				UpperCAmelCase__							(			self      : Optional[int]							,    lowerCamelCase_      : Optional[Any]							,    lowerCamelCase_      : Tuple							,    lowerCamelCase_      : Optional[Any]							,    lowerCamelCase_      : Optional[int]	)    -> List[str]:
								__magic_name__				:						Optional[int]     				=				20
								__magic_name__				:						Optional[Any]     				=				model_class_name(lowerCamelCase_	)

								__magic_name__				:						Union[str, Any]     				=				jnp.concatenate(
								    [attention_mask, jnp.zeros((attention_mask.shape[0], max_decoder_length - attention_mask.shape[1])	)]							,    axis=-1							,    )

								__magic_name__				:						int     				=				model.init_cache(input_ids.shape[0]							,    lowerCamelCase_	)
								__magic_name__				:						Optional[int]     				=				jnp.broadcast_to(
								    jnp.arange(input_ids.shape[-1] - 1	)[None, :]							,    (input_ids.shape[0], input_ids.shape[-1] - 1)	)

								__magic_name__				:						Union[str, Any]     				=				model(
								    input_ids[:, :-1]							,    attention_mask=lowerCamelCase_							,    past_key_values=lowerCamelCase_							,    position_ids=lowerCamelCase_							,    )
								__magic_name__				:						Tuple     				=				jnp.array(input_ids.shape[0] * [[input_ids.shape[-1] - 1]]							,    dtype='''i4'''	)
								__magic_name__				:						Optional[int]     				=				model(
								    input_ids[:, -1:]							,    past_key_values=outputs_cache.past_key_values							,    attention_mask=lowerCamelCase_							,    position_ids=lowerCamelCase_							,    )

								__magic_name__				:						int     				=				model(lowerCamelCase_							,    attention_mask=lowerCamelCase_	)
								__magic_name__				:						List[Any]     				=				np.max(np.abs((outputs_cache_next[0][:, -1, :5] - outputs[0][:, -1, :5])	)	)
								self.parent.assertTrue(diff < 1E-3							,    msg=F'''Max diff is {diff}'''	)






@require_sentencepiece
@require_flax
class  lowerCamelCase     (		_lowerCamelCase					,_lowerCamelCase					,unittest.TestCase    ):






			'''simple docstring'''





			UpperCamelCase__										=(FlaxXGLMModel, FlaxXGLMForCausalLM) if is_flax_available() else ()
			UpperCamelCase__										=(FlaxXGLMForCausalLM,) if is_flax_available() else ()

			def 				UpperCAmelCase__							(			self      : str	)    -> Tuple:
								__magic_name__				:						Tuple     				=				FlaxXGLMModelTester(self	)

			def 				UpperCAmelCase__							(			self      : Dict	)    -> List[str]:
								for model_class_name in self.all_model_classes:
													__magic_name__				,     __magic_name__				,     __magic_name__				:						Any     				=				self.model_tester.prepare_config_and_inputs()
													self.model_tester.check_use_cache_forward(lowerCamelCase_							,    lowerCamelCase_							,    lowerCamelCase_							,    lowerCamelCase_	)

			def 				UpperCAmelCase__							(			self      : List[str]	)    -> int:
								for model_class_name in self.all_model_classes:
													__magic_name__				,     __magic_name__				,     __magic_name__				:						int     				=				self.model_tester.prepare_config_and_inputs()
													self.model_tester.check_use_cache_forward_with_attn_mask(
													    lowerCamelCase_							,    lowerCamelCase_							,    lowerCamelCase_							,    lowerCamelCase_	)

			@slow
			def 				UpperCAmelCase__							(			self      : Any	)    -> Optional[Any]:
								__magic_name__				:						List[str]     				=				XGLMTokenizer.from_pretrained('''XGLM'''							,    padding_side='''left'''	)
								__magic_name__				:						Tuple     				=				tokenizer(['''Hello this is a long string''', '''Hey''']							,    return_tensors='''np'''							,    padding=lowerCamelCase_							,    truncation=lowerCamelCase_	)

								__magic_name__				:						Dict     				=				FlaxXGLMForCausalLM.from_pretrained('''facebook/xglm-564M'''	)
								__magic_name__				:						Optional[Any]     				=				1
								__magic_name__				:						str     				=				False

								__magic_name__				:						List[str]     				=				jax.jit(model.generate	)

								__magic_name__				:						List[Any]     				=				jit_generate(inputs['''input_ids''']							,    attention_mask=inputs['''attention_mask''']	).sequences

								__magic_name__				:						Optional[Any]     				=				tokenizer.batch_decode(lowerCamelCase_							,    skip_special_tokens=lowerCamelCase_	)

								__magic_name__				:						Dict     				=				[
								    '''Hello this is a long string of questions, but I\'m not sure if I\'m''',
								    '''Hey, I\'m a newbie to the forum and I\'''',
								]

								self.assertListEqual(lowerCamelCase_							,    lowerCamelCase_	)

			@is_pt_flax_cross_test
			def 				UpperCAmelCase__							(			self      : Any	)    -> Union[str, Any]:
								__magic_name__				,     __magic_name__				:						List[str]     				=				self.model_tester.prepare_config_and_inputs_for_common()

								for model_class in self.all_model_classes:
													with self.subTest(model_class.__name__	):
																		# prepare inputs
																		__magic_name__				:						List[Any]     				=				self._prepare_for_class(lowerCamelCase_							,    lowerCamelCase_	)
																		__magic_name__				:						Tuple     				=				{k: torch.tensor(v.tolist()	) for k, v in prepared_inputs_dict.items()}

																		# load corresponding PyTorch class
																		__magic_name__				:						Optional[int]     				=				model_class.__name__[4:]  # Skip the "Flax" at the beginning
																		__magic_name__				:						Optional[int]     				=				getattr(lowerCamelCase_							,    lowerCamelCase_	)

																		__magic_name__				,     __magic_name__				:						Optional[Any]     				=				pt_inputs['''input_ids'''].shape
																		__magic_name__				:						int     				=				np.random.randint(0							,    seq_length - 1							,    size=(batch_size,)	)
																		for batch_idx, start_index in enumerate(lowerCamelCase_	):
																							__magic_name__				:						Optional[int]     				=				0
																							__magic_name__				:						Optional[Any]     				=				1
																							__magic_name__				:						Optional[int]     				=				0
																							__magic_name__				:						str     				=				1
																		__magic_name__				:						List[str]     				=				pt_model_class(lowerCamelCase_	).eval()
																		# Flax models don't use the `use_cache` option and cache is not returned as a default.
																		# So we disable `use_cache` here for PyTorch model.
																		__magic_name__				:						str     				=				False
																		__magic_name__				:						Any     				=				model_class(lowerCamelCase_							,    dtype=jnp.floataa	)
																		__magic_name__				:						Optional[int]     				=				convert_pytorch_state_dict_to_flax(pt_model.state_dict()							,    lowerCamelCase_	)
																		__magic_name__				:						int     				=				fx_state

																		with torch.no_grad():
																							__magic_name__				:						Tuple     				=				pt_model(**lowerCamelCase_	).to_tuple()

																		__magic_name__				:						Optional[int]     				=				fx_model(**lowerCamelCase_	).to_tuple()
																		self.assertEqual(len(lowerCamelCase_	)							,    len(lowerCamelCase_	)							,    '''Output lengths differ between Flax and PyTorch'''	)
																		for fx_output, pt_output in zip(lowerCamelCase_							,    lowerCamelCase_	):
																							self.assert_almost_equals(fx_output[:, -1]							,    pt_output[:, -1].numpy()							,    4E-2	)

																		with tempfile.TemporaryDirectory() as tmpdirname:
																							pt_model.save_pretrained(lowerCamelCase_	)
																							__magic_name__				:						List[str]     				=				model_class.from_pretrained(lowerCamelCase_							,    from_pt=lowerCamelCase_	)

																		__magic_name__				:						Optional[int]     				=				fx_model_loaded(**lowerCamelCase_	).to_tuple()
																		self.assertEqual(
																		    len(lowerCamelCase_	)							,    len(lowerCamelCase_	)							,    '''Output lengths differ between Flax and PyTorch'''	)
																		for fx_output_loaded, pt_output in zip(lowerCamelCase_							,    lowerCamelCase_	):
																							self.assert_almost_equals(fx_output_loaded[:, -1]							,    pt_output[:, -1].numpy()							,    4E-2	)

			@is_pt_flax_cross_test
			def 				UpperCAmelCase__							(			self      : int	)    -> List[str]:
								__magic_name__				,     __magic_name__				:						Tuple     				=				self.model_tester.prepare_config_and_inputs_for_common()
								for model_class in self.all_model_classes:
													with self.subTest(model_class.__name__	):
																		# prepare inputs
																		__magic_name__				:						List[Any]     				=				self._prepare_for_class(lowerCamelCase_							,    lowerCamelCase_	)
																		__magic_name__				:						Any     				=				{k: torch.tensor(v.tolist()	) for k, v in prepared_inputs_dict.items()}

																		# load corresponding PyTorch class
																		__magic_name__				:						Optional[int]     				=				model_class.__name__[4:]  # Skip the "Flax" at the beginning
																		__magic_name__				:						int     				=				getattr(lowerCamelCase_							,    lowerCamelCase_	)

																		__magic_name__				:						str     				=				pt_model_class(lowerCamelCase_	).eval()
																		__magic_name__				:						Any     				=				False
																		__magic_name__				:						Any     				=				model_class(lowerCamelCase_							,    dtype=jnp.floataa	)

																		__magic_name__				:						str     				=				load_flax_weights_in_pytorch_model(lowerCamelCase_							,    fx_model.params	)
																		__magic_name__				,     __magic_name__				:						str     				=				pt_inputs['''input_ids'''].shape
																		__magic_name__				:						int     				=				np.random.randint(0							,    seq_length - 1							,    size=(batch_size,)	)
																		for batch_idx, start_index in enumerate(lowerCamelCase_	):
																							__magic_name__				:						str     				=				0
																							__magic_name__				:						str     				=				1
																							__magic_name__				:						Tuple     				=				0
																							__magic_name__				:						List[Any]     				=				1

																		# make sure weights are tied in PyTorch
																		pt_model.tie_weights()

																		with torch.no_grad():
																							__magic_name__				:						Any     				=				pt_model(**lowerCamelCase_	).to_tuple()

																		__magic_name__				:						Dict     				=				fx_model(**lowerCamelCase_	).to_tuple()
																		self.assertEqual(len(lowerCamelCase_	)							,    len(lowerCamelCase_	)							,    '''Output lengths differ between Flax and PyTorch'''	)
																		for fx_output, pt_output in zip(lowerCamelCase_							,    lowerCamelCase_	):
																							self.assert_almost_equals(fx_output[:, -1]							,    pt_output[:, -1].numpy()							,    4E-2	)

																		with tempfile.TemporaryDirectory() as tmpdirname:
																							fx_model.save_pretrained(lowerCamelCase_	)
																							__magic_name__				:						Union[str, Any]     				=				pt_model_class.from_pretrained(lowerCamelCase_							,    from_flax=lowerCamelCase_	)

																		with torch.no_grad():
																							__magic_name__				:						List[Any]     				=				pt_model_loaded(**lowerCamelCase_	).to_tuple()

																		self.assertEqual(
																		    len(lowerCamelCase_	)							,    len(lowerCamelCase_	)							,    '''Output lengths differ between Flax and PyTorch'''	)
																		for fx_output, pt_output in zip(lowerCamelCase_							,    lowerCamelCase_	):
																							self.assert_almost_equals(fx_output[:, -1]							,    pt_output[:, -1].numpy()							,    4E-2	)


			@slow
			def 				UpperCAmelCase__							(			self      : Dict	)    -> List[str]:
								for model_class_name in self.all_model_classes:
													__magic_name__				:						Union[str, Any]     				=				model_class_name.from_pretrained('''facebook/xglm-564M'''	)
													__magic_name__				:						Optional[int]     				=				model(np.ones((1, 1)	)	)
													self.assertIsNotNone(lowerCamelCase_	)



