Kirill 0532025b04 Fix llama 13B weights loading (#700) ... * Fix llama 13B weights loading * refactor more * add test * test storage offset * fix spacing * fix strides * llama 13B working? * yolo? * better test for seeks		2023-03-15 08:59:52 -07:00
..
external	err, external_test_opt.py broke...fusing will have to wait. correctness over speed	2023-03-11 17:54:47 -08:00
extra	Fix llama 13B weights loading (#700)	2023-03-15 08:59:52 -07:00
models	multistep optim tests passing	2023-03-11 17:49:53 -08:00
unit	Move rawbuffer (#697)	2023-03-13 22:30:36 -07:00
__init__.py	All devices are equal! (#196)	2020-12-15 23:44:08 -08:00
Dockerfile	testdocker	2023-01-09 12:41:52 -08:00
test_assign.py	assign buffer reuse (#547)	2023-02-09 11:53:02 -06:00
test_conv.py	Add conv binops_no_rerun test assertions (#665)	2023-03-10 19:09:48 -08:00
test_conv_shapetracker.py	speed like you wouldn't believe (#626)	2023-03-02 07:49:19 -08:00
test_custom_function.py	compile works (#688)	2023-03-12 11:01:25 -07:00
test_dtype.py	fix float16 in CLANG on linux	2023-03-11 21:51:22 -08:00
test_gc.py	fix up eye and fix gc test	2023-02-27 06:53:18 -08:00
test_jit.py	fix shapetracker test	2023-03-12 22:33:25 -07:00
test_net_speed.py	rename log_softmax, support dim, fix onnx Softmax	2023-02-24 10:11:24 -08:00
test_nn.py	relax atol, merge_view	2023-03-03 07:48:44 -08:00
test_ops.py	fix flip bug, add new unit tests	2023-03-12 23:55:31 -07:00
test_optim.py	multistep optim tests passing	2023-03-11 17:49:53 -08:00
test_randomness.py	Add tests for randomness (#621)	2023-03-01 15:39:20 -08:00
test_specific_conv.py	clean up opt (#649)	2023-03-05 20:49:36 -08:00
test_speed_v_torch.py	mem_estimate tracks bytes, not items	2023-03-10 09:44:12 -08:00
test_tensor.py	ConvNeXt + extras (#657)	2023-03-06 22:10:56 -08:00