From 5d90b38c4d280272106ad656808b35ff75bd46a0 Mon Sep 17 00:00:00 2001 From: Titus von Koeller Date: Wed, 27 Jul 2022 21:16:04 -0700 Subject: adding CLI tool for CUDA install debugging - intermediate commit --- setup.py | 3 +++ 1 file changed, 3 insertions(+) (limited to 'setup.py') 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', -- cgit v1.2.3