6 Commits

Author	SHA1	Message	Date
Christopher Mauri Milan	7f01dd04f0	Apply ruff linting rules to tests (#2473) ... * everything except F821 * enable F821 with noqa * dumb fix * fix remaining imports and (former) lambdas * replace _ with noqa to avoid gc	2023-11-27 21:24:06 -08:00
George Hotz	095e2ced61	add name support to fetch (#2407) ... * add name support * use fetch in gpt2 * remove requests from main lib, networkx also optional * umm, keep that assert * updates to fetch * i love the walrus so much * stop bundling mnist with tinygrad * err, https * download cache names * add DOWNLOAD_CACHE_VERSION * need env. * ugh, wrong path * replace get_child	2023-11-23 14:16:17 -08:00
George Hotz	a0890f4e6c	move fetch to helpers (#2363) ... * switch datasets to new fetch * add test_helpers * fix convnext and delete old torch load	2023-11-19 12:29:51 -08:00
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