From 96b3e7d78914f5db043e8b9ae3b3f72498abca4e Mon Sep 17 00:00:00 2001 From: shunting314 Date: Thu, 18 Dec 2025 18:06:58 -0800 Subject: [PATCH] [ez] log why fail to gen golden ref (#170843) Sometimes when doing accuracy test with a large batch size, it fail due to the golden ref can not be generated. It would be helpful to log why we fail to generate the golden ref. In my case, it's due to OOM. Pull Request resolved: https://github.com/pytorch/pytorch/pull/170843 Approved by: https://github.com/malfet, https://github.com/Skylion007 --- benchmarks/dynamo/common.py | 1 + 1 file changed, 1 insertion(+) diff --git a/benchmarks/dynamo/common.py b/benchmarks/dynamo/common.py index 64335937c95..a295c80f774 100644 --- a/benchmarks/dynamo/common.py +++ b/benchmarks/dynamo/common.py @@ -2216,6 +2216,7 @@ class BenchmarkRunner: log.warning( "fp64 golden ref were not generated for %s. Setting accuracy check to cosine", name, + exc_info=True, ) self.args.cosine = True fp64_outputs = None