20 Commits

Author	SHA1	Message	Date
Tim Dettmers	758c7175a2	Merge branch 'debug' into cuda-bin-switch-and-cli	2022-08-04 08:03:00 -07:00
Tim Dettmers	ab72a1294f	Added pre/post device call for extract outliers.	2022-08-04 07:47:22 -07:00
Tim Dettmers	cc5b323876	Merge branch 'extract_outliers' into debug	2022-08-04 07:40:48 -07:00
Tim Dettmers	6101a8fb9f	Added pre and post device call to transform.	2022-08-04 07:28:12 -07:00
Tim Dettmers	320eacb4c2	Removed print statement.	2022-08-03 14:17:54 -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
Tim Dettmers	2f01865a2f	Added CUDA block assert and is_on_gpu check.	2022-08-03 09:05:37 -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	5737f2b027	Merge branch 'patch_merge' into extract_outliers	2022-07-26 19:38:01 -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	c771b3a75a	Most tests passing.	2022-07-22 14:41:05 -07:00
Max Ryabinin	e4cf33f2a3	Fix imports	2022-07-01 17:25:44 +03:00
Max Ryabinin	8258b4364a	Add a CPU-only build option	2022-07-01 17:16:10 +03:00
Max Ryabinin	33efe4a09f	Remove unused imports, fix NotImplementedError	2022-06-30 18:14:20 +03:00
Tim Dettmers	8b3c0f355c	Added adagrad with tests (no clipping).	2021-11-10 15:10:02 -08:00
Tim Dettmers	a6eae2e7f2	Added skip_zeros; tests are passing.	2021-10-20 19:15:47 -07: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