Skip to content

Commit 42317cf

Browse files
Merge pull request #44 from codeflash-ai/tracer_tracker
Multi-threading support for the tracer
2 parents 76df611 + 2f4bdd9 commit 42317cf

File tree

5 files changed

+359
-139
lines changed

5 files changed

+359
-139
lines changed

code_to_optimize/code_directories/simple_tracer_e2e/workload.py

Lines changed: 10 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,4 @@
1+
from concurrent.futures import ThreadPoolExecutor
12
def funcA(number):
23
k = 0
34
for i in range(number * 100):
@@ -8,7 +9,14 @@ def funcA(number):
89
# Use a generator expression directly in join for more efficiency
910
return " ".join(str(i) for i in range(number))
1011

12+
def test_threadpool() -> None:
13+
pool = ThreadPoolExecutor(max_workers=3)
14+
args = list(range(10, 31, 10))
15+
result = pool.map(funcA, args)
16+
17+
for r in result:
18+
print(r)
19+
1120

1221
if __name__ == "__main__":
13-
for i in range(10, 31, 10):
14-
funcA(10)
22+
test_threadpool()

0 commit comments

Comments
 (0)