Allow TF32 in CUDA for increased performance #279

This commit is contained in:
AUTOMATIC
2022-09-12 16:34:13 +03:00
parent 11e648f6c7
commit b70b51cc72
2 changed files with 21 additions and 0 deletions

10
modules/errors.py Normal file
View File

@@ -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)