commit c68fac9c25c4bcbfdc627567313684582ef9ad55
author Nicolas Macchioni <nmacchioni@meta.com> Fri Jul 07 21:47:35 2023 +0000
committer PyTorch MergeBot <pytorchmergebot@users.noreply.github.com> Fri Jul 07 21:47:35 2023 +0000
tree aea602403d7e2af934e80457a4f5291173fcd8f8
parent ed4a8869afa0e9b28848adb6035ecc85638dec7e

[pt2][inductor] include `allow_tf32` in system information (#104129)

Summary: include `allow_tf32` in system information; previously aten results did not specify whether `allow_tf32` was true or not

Test Plan: sandcastle + CI

Differential Revision: D46568468

Pull Request resolved: https://github.com/pytorch/pytorch/pull/104129
Approved by: https://github.com/jansel

torch/_inductor/codecache.py

diff --git a/torch/_inductor/codecache.py b/torch/_inductor/codecache.py
index d4808f9..2fd8add 100644
--- a/torch/_inductor/codecache.py
+++ b/torch/_inductor/codecache.py
@@ -120,6 +120,9 @@
                 "cuda": torch.version.cuda,
                 "triton": triton_version,
             },
+            "other": {
+                "allow_tf32": torch.backends.cuda.matmul.allow_tf32,
+            },
         }
 
         system["hash"] = hashlib.sha256(