This commit is contained in:
PawelDopierala 2024-11-26 22:08:16 +01:00
parent d6fa089f1e
commit 84aa2a0637

12
main.py
View File

@ -33,11 +33,13 @@ async def compute(data: ComputeInput):
ranges = [(a_start + i * chunk_size, min(a_start + (i + 1) * chunk_size - 1, b_end)) ranges = [(a_start + i * chunk_size, min(a_start + (i + 1) * chunk_size - 1, b_end))
for i in range(num_chunks)] for i in range(num_chunks)]
with ThreadPoolExecutor() as executor: return ranges
futures = [executor.submit(run_compute, a, b, sigma) for a, b in ranges] #
results = [f.result() for f in futures] # with ThreadPoolExecutor() as executor:
# futures = [executor.submit(run_compute, a, b, sigma) for a, b in ranges]
return sum(results) % 999999937 # results = [f.result() for f in futures]
#
# return sum(results) % 999999937
if __name__ == "__main__": if __name__ == "__main__":
uvicorn.run(app, host="0.0.0.0", port=8000) uvicorn.run(app, host="0.0.0.0", port=8000)