Skip to content

Commit 4d439a6

Browse files
committed
Add more info to elemwise_openmp_speedup
1 parent b9792d8 commit 4d439a6

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

pytensor/misc/elemwise_openmp_speedup.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -64,6 +64,9 @@ def runScript(N):
6464
costlySpeed = costlyTimeOpenmp / costlyTime
6565
costlySpeedstring = "slowdown"
6666
print(f"Timed with vector of {int(options.N)} elements")
67+
print(
68+
f"{config.openmp_elemwise_minsize=}, OMP_NUM_THREADS={os.getenv('OMP_NUM_THREADS')}"
69+
)
6770
print(
6871
f"Fast op time without openmp {cheapTime}s with openmp {cheapTimeOpenmp}s {cheapSpeedstring} {cheapSpeed:2.2f}"
6972
)

0 commit comments

Comments
 (0)