JeffreyXiang commited on
Commit
1aa888c
·
1 Parent(s): 691d6b0
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -5,6 +5,7 @@ import os
5
  os.environ["OPENCV_IO_ENABLE_OPENEXR"] = '1'
6
  os.environ["PYTORCH_CUDA_ALLOC_CONF"] = "expandable_segments:True"
7
  os.environ["FLEX_GEMM_AUTOTUNE_CACHE_PATH"] = os.path.join(os.path.dirname(os.path.abspath(__file__)), 'autotune_cache.json')
 
8
  from datetime import datetime
9
  import shutil
10
  import cv2
 
5
  os.environ["OPENCV_IO_ENABLE_OPENEXR"] = '1'
6
  os.environ["PYTORCH_CUDA_ALLOC_CONF"] = "expandable_segments:True"
7
  os.environ["FLEX_GEMM_AUTOTUNE_CACHE_PATH"] = os.path.join(os.path.dirname(os.path.abspath(__file__)), 'autotune_cache.json')
8
+ os.environ["FLEX_GEMM_AUTOTUNER_VERBOSE"] = '1'
9
  from datetime import datetime
10
  import shutil
11
  import cv2