4 Commits

Author	SHA1	Message	Date
Tri Dao	0e8c46ae08	Run isort and black on test files	2023-08-18 20:59:35 -07:00
Tri Dao	88173a1aaf	[FusedDense] Support relu, rename FusedDenseGeluDense -> FusedMLP	2023-01-17 18:12:27 -08:00
Tri Dao	e68ebbe89a	Simplify FusedDense	2022-12-22 21:25:31 -08:00
Tri Dao	fa6d1ce44f	Add fused_dense and dropout_add_layernorm CUDA extensions	2022-11-13 21:59:20 -08:00