Skip to content

Commit 488670e

Browse files
committed
Trust input in elemwise_time_test
1 parent 4d439a6 commit 488670e

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

pytensor/misc/elemwise_time_test.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -49,6 +49,9 @@ def ElemwiseOpTime(N, script=False, loops=1000):
4949
v = np.random.random(N).astype(config.floatX)
5050
f = pytensor.function([x], 2 * x + x * x)
5151
f1 = pytensor.function([x], tanh(x))
52+
# Don't waste time filtering inputs
53+
f.trust_input = True
54+
f1.trust_input = True
5255
if not script:
5356
if config.openmp:
5457
print("With openmp:")

0 commit comments

Comments
 (0)