4 Commits

Author	SHA1	Message	Date
Laurent Mazare	dc47224ab9	Override the default cudnn heuristics. (#957)	2023-09-25 10:31:53 +01:00
Laurent Mazare	7b1f2da828	Cudnn fix. (#758)	2023-09-06 17:39:39 +01:00
Laurent Mazare	a044907ffc	Dilated convolutions (#657) ... * Add the dilation parameter. * Restore the basic optimizer example. * Dilation support in cudnn. * Use the dilation parameter in the cpu backend. * More dilation support. * No support for dilation in transposed convolutions. * Add dilation to a test. * Remove a print. * Helper function.	2023-08-29 16:12:11 +01:00
Laurent Mazare	90374097dc	Cudnn support (#445) ... * Add a cudnn feature to be used for conv2d. * Allocate the proper workspace. * Only create a single cudnn handle per cuda device. * Proper cudnn usage. * Bugfix.	2023-08-14 21:30:41 +01:00