From b70b51cc7248f57dcf16add86701139432c21e5b Mon Sep 17 00:00:00 2001
From: AUTOMATIC <16777216c@gmail.com>
Date: Mon, 12 Sep 2022 16:34:13 +0300
Subject:  Allow TF32 in CUDA for increased performance #279

---
 modules/errors.py | 10 ++++++++++
 1 file changed, 10 insertions(+)
 create mode 100644 modules/errors.py

(limited to 'modules/errors.py')

diff --git a/modules/errors.py b/modules/errors.py
new file mode 100644
index 00000000..372dc51a
--- /dev/null
+++ b/modules/errors.py
@@ -0,0 +1,10 @@
+import sys
+import traceback
+
+
+def run(code, task):
+    try:
+        code()
+    except Exception as e:
+        print(f"{task}: {type(e).__name__}", file=sys.stderr)
+        print(traceback.format_exc(), file=sys.stderr)
-- 
cgit v1.2.3