summaryrefslogtreecommitdiff
path: root/setup.py
diff options
context:
space:
mode:
authorTitus von Koeller <titus@vonkoeller.com>2022-07-27 21:16:04 -0700
committerTitus von Koeller <titus@vonkoeller.com>2022-07-27 21:16:04 -0700
commit5d90b38c4d280272106ad656808b35ff75bd46a0 (patch)
tree993fc302f0ea6890ed46dcc9c3de28815721d9f8 /setup.py
parentbd515328d70f344f935075f359c5aefc616878d5 (diff)
adding CLI tool for CUDA install debugging - intermediate commit
Diffstat (limited to 'setup.py')
-rw-r--r--setup.py3
1 files changed, 3 insertions, 0 deletions
diff --git a/setup.py b/setup.py
index 6275ddd..3292e30 100644
--- a/setup.py
+++ b/setup.py
@@ -23,6 +23,9 @@ setup(
keywords="gpu optimizers optimization 8-bit quantization compression",
url="http://packages.python.org/bitsandbytes",
packages=find_packages(),
+ entry_points={
+ "console_scripts": ["debug_cuda = bitsandbytes.debug_cli:cli"],
+ },
package_data={'': ['libbitsandbytes.so']},
long_description=read('README.md'),
long_description_content_type='text/markdown',