chore: address review comments

pytorch · Oct 17, 2024 · b685087 · b685087
1 parent 97867f5
commit b685087
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/py/torch_tensorrt/dynamo/lowering/passes/accumulate_fp32_matmul.py b/py/torch_tensorrt/dynamo/lowering/passes/accumulate_fp32_matmul.py
@@ -55,6 +55,11 @@ def accumulate_fp32_matmul(
 
         gm = clean_up_graph_after_modifications(gm)
         logger.debug(
-            f"Graph after changing matmul layers to use FP32 accumulation:\n{gm.graph}"
+            f"Graph after enabling matmul layers to use FP32 accumulation:\n{gm.graph}"
         )
-        return gm
+    else:
+        logger.debug(
+            "Skipping FP32 accumulation for matmul layers as use_fp32_acc is not enabled in the compilation settings"
+        )
+
+    return gm