4147 Commits

This Branch

This Branch

All Branches

Author	SHA1	Message	Date
George Hotz	33f18c61a1	test_broadcasted_add	2022-06-12 10:19:58 -07:00
George Hotz	d47a421970	add cout to conv_args, don't change the first 12	2022-06-12 00:10:15 -07:00
George Hotz	af300b121b	refactor to pass conv args into llops	2022-06-11 23:08:46 -07:00
George Hotz	d747a4b9e2	add padding to conv2d function, other minor things	2022-06-11 22:29:42 -07:00
George Hotz	9a3c048724	skip broken tests, no float64 allowed	2022-06-11 17:12:04 -07:00
George Hotz	3461e1a467	A is actually a unary op	2022-06-11 17:05:29 -07:00
George Hotz	296f391403	delete old graph engine to reach line count	2022-06-11 17:02:53 -07:00
George Hotz	a4d0d3f17a	stupid numpy hack	2022-06-11 17:01:56 -07:00
George Hotz	c03c835d75	oops	2022-06-11 16:53:34 -07:00
George Hotz	361123490a	remove a lot of useless returns	2022-06-11 16:50:06 -07:00
George Hotz	35e55afe17	and processing op	2022-06-11 16:46:38 -07:00
George Hotz	6d5591f7a3	movement op to SSA	2022-06-11 16:44:24 -07:00
George Hotz	6685807df7	reduce_op in SSA format too	2022-06-11 16:40:14 -07:00
George Hotz	bbf231da34	move unary and binary op mem alloc to Ops class	2022-06-11 16:35:03 -07:00
George Hotz	1511cbf9c5	graphs of llops, sorry about the line count	2022-06-11 16:28:39 -07:00
George Hotz	9ebd472375	move ops to ops.py	2022-06-11 15:58:56 -07:00
George Hotz	b5b68e75ff	simpler onnx	2022-06-11 15:35:45 -07:00
George Hotz	2305a5347b	test_onnx works with enet also	2022-06-11 14:30:26 -07:00
George Hotz	6fdb276886	flip batchnorm function order	2022-06-11 13:20:41 -07:00
George Hotz	85d17a2acd	running resnet onnx	2022-06-11 13:17:15 -07:00
George Hotz	0225360191	fixed with one return x	2022-06-11 12:08:53 -07:00
George Hotz	094b304348	very weird += is broken	2022-06-11 11:58:50 -07:00
George Hotz	db5a632e8c	multicat + test onnx is generic onnx	2022-06-11 11:50:47 -07:00
George Hotz	a710b3a210	it's a real test now	2022-06-11 11:33:33 -07:00
George Hotz	8440dbfa5d	support inputs	2022-06-11 11:21:45 -07:00
George Hotz	08de1aa636	add flatten to tinygrad	2022-06-11 11:15:16 -07:00
George Hotz	aee251cc41	op model test	2022-06-11 11:06:03 -07:00
George Hotz	d061ce8d5e	add ELU support	2022-06-11 10:47:23 -07:00
George Hotz	50ba554a14	Ops layer of indirection	2022-06-11 10:04:45 -07:00
George Hotz	1d29780b75	use conv2d for transpose when able	2022-06-11 09:32:13 -07:00
George Hotz	861323c121	cleanup convdw torch	2022-06-11 08:32:57 -07:00
George Hotz	041ad55ed1	clean up torch processing op	2022-06-11 08:27:09 -07:00
George Hotz	fc7eabb86f	processing op	2022-06-11 08:12:02 -07:00
George Hotz	72186ebd5a	movement ops, reshape is a copy now	2022-06-10 20:01:47 -07:00
George Hotz	e5d694490f	reshape makes a copy	2022-06-10 19:49:04 -07:00
George Hotz	c8bacd0d8e	rename transpose to permute	2022-06-10 19:41:50 -07:00
George Hotz	462f1ce0da	Remove Matmul (#323)	2022-06-10 19:26:23 -07:00
George Hotz	66cb86c9ec	fix strided conv	2022-06-10 15:17:53 -07:00
George Hotz	8864b37333	fix torch convdw	2022-06-10 15:04:39 -07:00
George Hotz	aac1a9b419	this breaks tests	2022-06-10 12:20:42 -07:00
George Hotz	c4238187be	oops, we should have a linter in a precommit hook	2022-06-10 10:35:50 -07:00
George Hotz	160f651c25	conv_args not in llops interface	2022-06-10 10:35:10 -07:00
George Hotz	8c084b8c12	dispatcher: now this is nice	2022-06-09 10:28:46 -07:00
George Hotz	70bb3a7976	remove more weird logic in the dispatcher	2022-06-09 10:06:01 -07:00
George Hotz	e58d344759	simpler dispatch logic since the mlops are universal now	2022-06-09 10:01:10 -07:00
George Hotz	273af8d732	remove unused CPUBuffer functions	2022-06-09 09:45:39 -07:00
George Hotz	9d85067ad8	no splatting conv_args	2022-06-09 09:40:58 -07:00
George Hotz	259b536e3a	clean up reduce a little	2022-06-09 09:34:19 -07:00
George Hotz	d3e7238bdd	remove dead code	2022-06-09 09:25:40 -07:00
George Hotz	40f8eb7383	namedtuple for conv_args	2022-06-09 09:19:52 -07:00