6 Commits

Author	SHA1	Message	Date
Nikhil Mehta	0b7d71fd2f	Add softmin, hardshrink, hardtanh (#1180) ... --------- Co-authored-by: Nikhil Mehta <nikmehta@tesla.com>	2024-06-04 15:48:18 -07:00
Awni Hannun	21623156a3	Reset peak memory (#1074) ... * reset peak memory * fix linux * nits in docs	2024-05-03 17:12:51 -07:00
Awni Hannun	771575d27b	Expose function to clear memory cache (#1032) ... * expose function to clear memory cache * fix linux build * fix metal tests	2024-04-24 16:48:51 -07:00
Awni Hannun	3d405fb3b1	Add synchronize function (#1006) ... * add synchronize function * fix linux * fix linux * fix and fix docs * fix test * try synchronize in stream destroy * synchronize works for both cpu and gpu	2024-04-22 08:25:46 -07:00
Awni Hannun	8a0677d56d	Shared events for synchronization + async eval (#998) ... * more async eval * fix rebase * try correct async eval * fix async * more tests for async eval * use shared events for synchronization * comment + cleanup * with autorelease pool * fix no metal build * fix compile * fix patch * don't eval if asyn evale'd * don't use is_evaled * comments * more multi stream tests * try and cleanup use of is_evaled * use a status flag	2024-04-17 06:16:02 -07:00
Awni Hannun	d5964a2710	bindings for memory info (#761) ... * bindings for memory info * update api * keep cache low if requested * fix default * nit in ops error	2024-03-01 19:51:58 -08:00