From 229ec554778e633a0ceaf58265427c1b3a5628dc Mon Sep 17 00:00:00 2001 From: HydrogenSulfate <490868991@qq.com> Date: Tue, 19 Aug 2025 19:15:31 +0800 Subject: [PATCH] skip warning if not cuda device --- python/paddle/optimizer/lbfgs.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/python/paddle/optimizer/lbfgs.py b/python/paddle/optimizer/lbfgs.py index ec0c2f635a3606..7591526b2f7e60 100644 --- a/python/paddle/optimizer/lbfgs.py +++ b/python/paddle/optimizer/lbfgs.py @@ -59,10 +59,13 @@ class _LbfgsStateDict(TypedDict): def check_tf32_override(): """Check and warn about TF32 acceleration status""" - if os.getenv("NVIDIA_TF32_OVERRIDE") != "0": # None or "1" + if ( + paddle.device.is_compiled_with_cuda() + and os.getenv("NVIDIA_TF32_OVERRIDE") != "0" + ): # None or "1" warnings.warn( "Warning! TF32 Tensor Cores are enabled by default on some NVIDIA GPUs for faster computation, " - "but may compromise numerical precision in specific cases, particularly with the L-BFGS optimizer. " + "but may compromise numerical precision in specific cases, particularly with the L-BFGS optimizer." "To disable it, set: NVIDIA_TF32_OVERRIDE=0" )