George Hotz d58586bb17 safetensors! (#903) ... * safetensors test * safe_save * load back with real safetensors * bugfix in device name. add simple torch_load * it works for llama, but it's slower... * mmap * no intermediate * load mmaped * readinto speed * not ready yet * revert that		2023-06-02 13:41:09 -07:00
..
external	Onnx ops And, Or, Xor, Not (#847)	2023-05-29 11:09:20 -07:00
extra	make tests faster + add onnx (#815)	2023-05-27 08:53:32 -07:00
models	Zero dim Tensor support (#777)	2023-06-01 11:32:02 -07:00
unit	safetensors! (#903)	2023-06-02 13:41:09 -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	Disktensors! (#819)	2023-05-28 15:40:37 -07:00
test_dtype.py	Llvm int support (#866)	2023-05-30 17:49:26 -07:00
test_gc.py	fix up eye and fix gc test	2023-02-27 06:53:18 -08:00
test_jit.py	assertions for jit	2023-05-05 21:56:32 -07:00
test_net_speed.py	test touchups	2023-05-26 02:39:42 +00:00
test_nn.py	support Int64 type in cstyle gen (#860)	2023-05-30 16:04:46 -07:00
test_ops.py	Fix .std() tests on torch=1.13 (#904)	2023-06-02 07:33:51 -07:00
test_optim.py	Add weight decay to SGD (#883)	2023-06-01 13:13:18 -07:00
test_randomness.py	add nn layer tests	2023-06-01 21:34:24 -07:00
test_specific_conv.py	disable that test on LLVM. i have to stop pushing to master	2023-05-26 03:11:03 +00:00
test_speed_v_torch.py	good changes from the M1 Tensor Core project (#730)	2023-03-29 05:11:02 +04:00
test_tensor.py	Zero dim Tensor support (#777)	2023-06-01 11:32:02 -07:00