From 6ed83974db3ff1de075331985efc2543eddb6656 Mon Sep 17 00:00:00 2001 From: Ricardo Vieira Date: Fri, 31 May 2024 15:14:55 +0200 Subject: [PATCH] Do not emmit confusing warning from FusionOptimizer by default --- pytensor/tensor/rewriting/elemwise.py | 10 ++++------ 1 file changed, 4 insertions(+), 6 deletions(-) diff --git a/pytensor/tensor/rewriting/elemwise.py b/pytensor/tensor/rewriting/elemwise.py index ac4918117e..2863d71d9d 100644 --- a/pytensor/tensor/rewriting/elemwise.py +++ b/pytensor/tensor/rewriting/elemwise.py @@ -699,12 +699,10 @@ def elemwise_scalar_op_has_c_code(node: Apply) -> bool: if node.op.scalar_op.supports_c_code(node.inputs, node.outputs): return True else: - warn( - "Optimization Warning: " - f"The Op {node.op.scalar_op} does not provide a C implementation." - " As well as being potentially slow, this also disables " - "loop fusion." - ) + if config.optimizer_verbose: + warn( + f"Loop fusion interrupted because {node.op.scalar_op} does not provide a C implementation." + ) return False # Fuseable nodes have to be accessed in a deterministic manner