From 8724c990c74c1bc4a7e2c97abb913015ca185a06 Mon Sep 17 00:00:00 2001
From: adpkadspokasdk <64224601+Cyberes@users.noreply.github.com>
Date: Thu, 27 Oct 2022 16:04:49 -0600
Subject: [PATCH] allow hiding of the welcome message

---
 bitsandbytes/cuda_setup/main.py | 21 ++++++++++++---------
 environment.yml                 |  1 +
 2 files changed, 13 insertions(+), 9 deletions(-)

diff --git a/bitsandbytes/cuda_setup/main.py b/bitsandbytes/cuda_setup/main.py
index 0fc813b..d83f2f5 100644
--- a/bitsandbytes/cuda_setup/main.py
+++ b/bitsandbytes/cuda_setup/main.py
@@ -17,6 +17,7 @@ evaluation:
 """
 
 import ctypes
+import os
 
 from .paths import determine_cuda_runtime_lib_path
 from bitsandbytes.cextension import CUDASetup
@@ -107,15 +108,17 @@ def get_compute_capability(cuda):
 
 
 def evaluate_cuda_setup():
-    # we remove this for now and see how things go
-    #print('')
-    #print('='*35 + 'BUG REPORT' + '='*35)
-    #print('Welcome to bitsandbytes. For bug reports, please submit your error trace to: https://github.com/TimDettmers/bitsandbytes/issues')
-    #print('For effortless bug reporting copy-paste your error into this form: https://docs.google.com/forms/d/e/1FAIpQLScPB8emS3Thkp66nvqwmjTEgxp8Y9ufuWTzFyr9kJ5AoI47dQ/viewform?usp=sf_link')
-    #print('='*80)
-    #if not torch.cuda.is_available():
-        #print('No GPU detected. Loading CPU library...')
-        #return binary_name
+    if 'BITSANDBYTES_NOWELCOME' not in os.environ or str(os.environ['BITSANDBYTES_NOWELCOME']) == '0':
+        print('')
+        print('=' * 35 + 'BUG REPORT' + '=' * 35)
+        print('Welcome to bitsandbytes. For bug reports, please submit your error trace to: https://github.com/TimDettmers/bitsandbytes/issues')
+        print('For effortless bug reporting copy-paste your error into this form: https://docs.google.com/forms/d/e/1FAIpQLScPB8emS3Thkp66nvqwmjTEgxp8Y9ufuWTzFyr9kJ5AoI47dQ/viewform?usp=sf_link')
+        print('To hide this message, set the BITSANDBYTES_NOWELCOME variable like so: export BITSANDBYTES_NOWELCOME=1')
+        print('=' * 80)
+
+    # if not torch.cuda.is_available():
+    #     print('No GPU detected. Loading CPU library...')
+    #     return binary_name
 
     binary_name = "libbitsandbytes_cpu.so"
 
diff --git a/environment.yml b/environment.yml
index 6bc6f9a..93f5b38 100644
--- a/environment.yml
+++ b/environment.yml
@@ -1,6 +1,7 @@
 name: 8-bit
 channels:
   - conda-forge
+  - pytorch
 dependencies:
   - python=3.9
   - pytest