diff --git a/execute.py b/execute.py index 8cbde9c..af78588 100644 --- a/execute.py +++ b/execute.py @@ -21,8 +21,6 @@ async def main(a_start, b_end, sigma): for i in range(num_chunks)] async with httpx.AsyncClient() as client: tasks = [send_request(client, server, rangee, sigma) for server, rangee in zip(ip_list, ranges)] - # tasks2 = [send_request(client, server, rangee, sigma) for server, rangee in zip(ip_list, ranges[16:32])] - # tasks.extend(tasks2) results = await asyncio.gather(*tasks) final_result = 0 for idx, result in enumerate(results): diff --git a/main.py b/main.py index e19608a..6ede326 100644 --- a/main.py +++ b/main.py @@ -33,13 +33,11 @@ async def compute(data: ComputeInput): ranges = [(a_start + i * chunk_size, min(a_start + (i + 1) * chunk_size - 1, b_end)) for i in range(num_chunks)] - return ranges - # - # with ThreadPoolExecutor() as executor: - # futures = [executor.submit(run_compute, a, b, sigma) for a, b in ranges] - # results = [f.result() for f in futures] - # - # return sum(results) % 999999937 + with ThreadPoolExecutor() as executor: + futures = [executor.submit(run_compute, a, b, sigma) for a, b in ranges] + results = [f.result() for f in futures] + + return sum(results) % 999999937 if __name__ == "__main__": uvicorn.run(app, host="0.0.0.0", port=8000)