[BugFix] fix log with paddlefleet.ops (#6528)

2026-04-23 17:11:21 +08:00 · 2026-02-27 14:34:29 +08:00
parent 1503443871
commit c6d8fbe526
1 changed files with 3 additions and 0 deletions
@@ -35,8 +35,11 @@ def load_deep_gemm():
            # SM100 should use PFCC DeepGemm
            paddle.compat.enable_torch_proxy(scope={"deep_gemm"})
            try:
+                import logging
+
                import paddlefleet.ops.deep_gemm as deep_gemm

+                logging.getLogger().handlers.clear()
                logger.info("Detected sm100, use PaddleFleet DeepGEMM")
            except:
                import deep_gemm as deep_gemm