"""simple docstring"""
import os
import time

import numpy as np
import onnxruntime as ort


SCREAMING_SNAKE_CASE__   =					"1"
SCREAMING_SNAKE_CASE__   =					"0"
SCREAMING_SNAKE_CASE__   =					"1"

SCREAMING_SNAKE_CASE__   =					ort.SessionOptions()
SCREAMING_SNAKE_CASE__   =					ort.GraphOptimizationLevel.ORT_DISABLE_ALL
print("Create inference session...")
SCREAMING_SNAKE_CASE__   =					["TensorrtExecutionProvider", "CUDAExecutionProvider"]
SCREAMING_SNAKE_CASE__   =					ort.InferenceSession("model.onnx", sess_options=sess_opt, providers=execution_provider)
SCREAMING_SNAKE_CASE__   =					ort.RunOptions()

SCREAMING_SNAKE_CASE__   =					128
SCREAMING_SNAKE_CASE__   =					1
SCREAMING_SNAKE_CASE__   =					np.ones((batch, sequence), dtype=np.intaa)
SCREAMING_SNAKE_CASE__   =					np.ones((batch, sequence), dtype=np.intaa)
SCREAMING_SNAKE_CASE__   =					np.ones((batch, sequence), dtype=np.intaa)

print("Warm up phase...")
sess.run(
    None,
    {
        sess.get_inputs()[0].name: input_ids,
        sess.get_inputs()[1].name: attention_mask,
        sess.get_inputs()[2].name: token_type_ids,
    },
    run_options=run_opt,
)

print("Start inference...")
SCREAMING_SNAKE_CASE__   =					time.time()
SCREAMING_SNAKE_CASE__   =					2_000
SCREAMING_SNAKE_CASE__   =					{}
for iter in range(max_iters):
					SCREAMING_SNAKE_CASE__   =					sess.run(
					    None,
					    {
					        sess.get_inputs()[0].name: input_ids,
					        sess.get_inputs()[1].name: attention_mask,
					        sess.get_inputs()[2].name: token_type_ids,
					    },
					    run_options=run_opt,
					)
print("Average Inference Time = {:.3f} ms".format((time.time() - start_time) * 1_000 / max_iters))



"""simple docstring"""
SCREAMING_SNAKE_CASE__   =					{str(digit): digit**5 for digit in range(10)}





def 				lowerCAmelCase__      (     _UpperCamelCase    :			int   )   ->			int:





						"""simple docstring"""






						return sum(DIGITS_FIFTH_POWER[digit] for digit in str(_UpperCamelCase   )   )





def 				lowerCAmelCase__      (     )   ->			int:





						"""simple docstring"""






						return sum(
						    number
						    for number in range(1_0_0_0	,     1_0_0_0_0_0_0   )
						    if number == digits_fifth_powers_sum(_UpperCamelCase   )   )


if __name__ == "__main__":
					print(solution())


