3 Commits

Author	SHA1	Message	Date
George Hotz	3baaf298d6	two stage cumsum in tensor.py (#2331) ... * two stage cumsum in tensor.py * 2 more kernels for llama cumsum * gpt-2 and llama use fast multinomial	2023-11-16 12:09:53 -08:00
chenyu	1b46de1a3e	fix type of helpers.prod, add test cases (#1859)	2023-09-14 05:16:55 +08:00
George Hotz	63c46e0287	Parens and gls (#1768) ... * more paren stripping * remove global and local size from renderers * complex strip parens * extra helpers + minor webgpu fix * fix test uops * one more parens test	2023-09-04 16:09:01 -07:00