提交 488670eb authored 作者: Ricardo Vieira's avatar Ricardo Vieira 提交者: Ricardo Vieira

Trust input in `elemwise_time_test`

上级 4d439a65
...@@ -49,6 +49,9 @@ def ElemwiseOpTime(N, script=False, loops=1000): ...@@ -49,6 +49,9 @@ def ElemwiseOpTime(N, script=False, loops=1000):
v = np.random.random(N).astype(config.floatX) v = np.random.random(N).astype(config.floatX)
f = pytensor.function([x], 2 * x + x * x) f = pytensor.function([x], 2 * x + x * x)
f1 = pytensor.function([x], tanh(x)) f1 = pytensor.function([x], tanh(x))
# Don't waste time filtering inputs
f.trust_input = True
f1.trust_input = True
if not script: if not script:
if config.openmp: if config.openmp:
print("With openmp:") print("With openmp:")
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论