18 Commits

Author	SHA1	Message	Date
Adrian Bakke	8a7bbcc240	Accel/cuda (#319) ... * start work * right a wrong * mimimal reduce example * save * get tests to pass * fix stupid bug * tests passing * out of bounds error fixed, all tests passing * revert changes to other files * clean * getting better, will drink a glass of wine and finish over the weekend * clean	2022-05-14 21:25:30 -07:00
George Hotz	0e6832a8ea	support torch GPU, only autoinit cuda in the buffer	2022-01-15 19:15:12 -08:00
George Hotz	52918fbf78	cuda stub	2022-01-15 19:02:17 -08:00
George Hotz	5efb6653c4	a bit of work on metal	2021-12-30 13:53:08 -05:00
George Hotz	b4839eb6bb	factor out binary_broadcast	2021-11-29 03:49:12 -05:00
George Hotz	ca154f8b59	add sub and a few refactors	2021-11-29 03:35:23 -05:00
George Hotz	7b3c442f3b	sum in metal	2021-11-29 03:13:41 -05:00
George Hotz	c6fb087695	fix old list behavior	2021-11-29 01:49:21 -05:00
George Hotz	7c160c6cee	metal: got relu working and add somewhat working	2021-11-29 01:40:18 -05:00
George Hotz	7d12482d80	refactor efficientnet loading	2021-10-30 17:02:17 -07:00
George Hotz	fc6597a6d9	only resnet18, it's too slow otherwise	2021-10-30 16:48:39 -07:00
George Hotz	c05867dcbb	header	2021-10-30 16:41:05 -07:00
George Hotz	e10dc90524	tpu docs	2021-10-30 16:38:19 -07:00
George Hotz	63f50cff45	move back again	2021-10-30 16:13:29 -07:00
George Hotz	3d646272d6	move back	2021-10-30 16:12:12 -07:00
George Hotz	ac8afd24fa	refactor accel	2021-10-30 16:10:59 -07:00
Wenjun.Zhang	5402fd7f05	Fix: (#279) ... 1. python create model not match swift model in ane build folder. 2. fix ane lib path and add a dummy backward function for use_ane.py can run in examples folder.	2021-08-06 07:58:13 -07:00
George Hotz	0eda3eb421	move ane and fpga to accel folder	2021-06-19 11:26:20 -07:00