75 Commits

Author	SHA1	Message	Date
Tim Dettmers	8f84674d67	Fixed bugs in cuda setup.	2022-08-04 09:16:00 -07:00
Tim Dettmers	758c7175a2	Merge branch 'debug' into cuda-bin-switch-and-cli	2022-08-04 08:03:00 -07:00
Tim Dettmers	cc5b323876	Merge branch 'extract_outliers' into debug	2022-08-04 07:40:48 -07:00
Tim Dettmers	451fd9506e	Added fixes for the case that matmullt dim A is zero, e.g. [0, 768].	2022-08-03 11:54:01 -07:00
Titus von Koeller	59a615b386	factored cuda_setup.main out into smaller modules and functions	2022-08-02 21:26:50 -07:00
Tim Dettmers	3479d02a76	Added some more docs and comments.	2022-08-01 19:43:09 -07:00
Tim Dettmers	8bf3e9faab	Added full env variable search; CONDA_PREFIX priority.	2022-08-01 19:22:41 -07:00
Titus von Koeller	ea7c14f8ef	reran black with linelength 80 for greater readability	2022-08-01 09:32:47 -07:00
Titus von Koeller	bfa0e33294	ran black and isort for coherent code formatting	2022-08-01 03:31:48 -07:00
Tim Dettmers	dd50382b32	Full evaluate_cuda setup with integration test.	2022-07-31 17:47:44 -07:00
Titus von Koeller	5d90b38c4d	adding CLI tool for CUDA install debugging - intermediate commit	2022-07-27 21:16:04 -07:00
Tim Dettmers	5737f2b027	Merge branch 'patch_merge' into extract_outliers	2022-07-26 19:38:01 -07:00
Tim Dettmers	32fa459ed7	Added col_ampere outlier extraction kernel.	2022-07-26 18:15:51 -07:00
Tim Dettmers	bcab99ec87	Working outlier extraction for Turing.	2022-07-26 17:39:30 -07:00
Tim Dettmers	cbb901ac51	Boilerplate and test for extract_outliers.	2022-07-26 12:12:38 -07:00
Tim Dettmers	1e88edd8c0	Removed rowscale (segfaults on ampere).	2022-07-25 17:27:57 -07:00
Tim Dettmers	8b1fd32e3e	Fixed makefile; fixed Ampere igemmlt_8 bug.	2022-07-25 14:02:14 -07:00
Tim Dettmers	c771b3a75a	Most tests passing.	2022-07-22 14:41:05 -07:00
Max Ryabinin	33efe4a09f	Remove unused imports, fix NotImplementedError	2022-06-30 18:14:20 +03:00
Tim Dettmers	20e1677dfd	Added module override, bnb.nn.Embedding #13 #15 #19	2021-11-29 09:32:13 -08:00
Tim Dettmers	108cf9fc1f	Fixed unsafe use of eval. #8	2021-11-29 08:21:05 -08:00
Tim Dettmers	2f8083bd8b	Added AdamW. #10 #13	2021-11-28 21:18:11 -08:00
Tim Dettmers	8b3c0f355c	Added adagrad with tests (no clipping).	2021-11-10 15:10:02 -08:00
Tim Dettmers	bb34fd50a1	Initial plumbing for skip_zeros.	2021-10-20 18:37:44 -07:00
Tim Dettmers	7439924891	Initial commit	2021-10-05 19:16:20 -07:00