George Hotz 03b38864db fix batchnorm at training (#753) ... * e2e testing * min failure * no affine on bn, still fails * why did i think i could detach that? * allow more kernels for bn * some test issue i don't understand		2023-04-19 08:01:04 -07:00
..
external	fix batchnorm at training (#753)	2023-04-19 08:01:04 -07:00
extra	linearizer! (#714)	2023-03-19 23:43:49 -07:00
models	fix batchnorm at training (#753)	2023-04-19 08:01:04 -07:00
unit	Remove Zeroview (#748)	2023-04-17 08:21:46 -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	fix binop, other tests failure (#723)	2023-03-22 18:15:07 -07:00
test_dtype.py	casting: support all backends and implement half (#726)	2023-03-24 09:58:03 -07: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	LayerNorm2d for 2 lines	2023-03-20 16:58:43 -07:00
test_ops.py	relu UnaryOp is back	2023-04-14 07:12:53 -07:00
test_optim.py	remove RMSprop, nobody uses it anymore	2023-03-20 12:31:34 -07: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	good changes from the M1 Tensor Core project (#730)	2023-03-29 05:11:02 +04:00
test_tensor.py	ConvNeXt + extras (#657)	2023-03-06 22:10:56 -08:00