George Hotz dc21e63bd2 test: put conv in one reduce (#4441) ... * test: put conv in one reduce * put reduce at the end * more expand * generic, and that expand was breaking things * ratio * don't undo the expand * arg 1 * strides * warning, for resnet * warning removed * disable cast * handle cast * op * err, that's right * fixup * fix that * a test to play with * add double_reduces * working up to final reshape * fold the last reshape * moved to schedule * fix axis * ci, need to bring arange back * FUSE_CONV_BW maybe * valid in 3.9 * test_expand_reduce_is_folded_on_different_axes * add FUSE_CONV_BW=1 * test_fold_batchnorm_backward * test_sgd_4convs_fuse --------- Co-authored-by: qazal <qazal.software@gmail.com>		2024-07-22 12:16:13 +03:00
..
external	test: put conv in one reduce (#4441)	2024-07-22 12:16:13 +03:00
imported	skip some redundant and slow tests in ci (#5416)	2024-07-12 14:43:13 -04:00
models	add failing regression test for image (#5540)	2024-07-17 17:27:18 -07:00
testextra	names shadowing builtins (#5179)	2024-06-27 08:15:01 -04:00
unit	one more test case for symbolic mod mul (#5615)	2024-07-20 17:23:06 -04:00
web	fast path for copy (#2548)	2023-12-01 11:34:47 -08:00
__init__.py	All devices are equal! (#196)	2020-12-15 23:44:08 -08:00
Dockerfile	Docker fix (#1039)	2023-06-25 10:38:58 -07:00
helpers.py	add bigint type for indexing [run_process_replay] (#5387)	2024-07-11 11:37:10 -07:00
test_arange.py	test: put conv in one reduce (#4441)	2024-07-22 12:16:13 +03:00
test_assign.py	test masked assign views (#4599)	2024-05-15 15:06:48 +03:00
test_const_folding.py	MetaOps.KERNEL (#5543)	2024-07-17 19:41:23 -07:00
test_conv.py	db in wal mode (#5388)	2024-07-12 20:43:36 -07:00
test_conv_shapetracker.py	test: put conv in one reduce (#4441)	2024-07-22 12:16:13 +03:00
test_copy_speed.py	remove cpu and torch backends (#3399)	2024-02-15 16:55:39 +01:00
test_custom_function.py	s/loadops/metaops [run_process_replay] (#5421)	2024-07-12 13:26:50 -07:00
test_device_speed.py	move uopgraph to file [run_process_replay] (#5364)	2024-07-10 17:34:50 -07:00
test_dtype.py	pretty print lazy op per default (#5505)	2024-07-18 09:34:08 -07:00
test_dtype_alu.py	elf loader (#5508)	2024-07-17 17:09:34 +03:00
test_fusion_op.py	process replay in all of CI (#4884)	2024-06-10 14:49:29 -04:00
test_fuzz_shape_ops.py	fix typing for test to run in py38 (#4930)	2024-06-12 13:22:30 -04:00
test_gc.py	threefry again (#3785)	2024-03-18 16:47:07 -04:00
test_graph.py	fix hcq sync (#5062)	2024-06-26 17:50:37 +03:00
test_hcq.py	hcq refactor signal into class (#5575)	2024-07-19 23:23:05 +03:00
test_image_dtype.py	add failing regression test for image (#5540)	2024-07-17 17:27:18 -07:00
test_jit.py	Approximations for SIN/LOG2/EXP2 passing all tests. (#5187)	2024-07-10 16:44:58 -07:00
test_kernel_cache.py	move the compiler cache to be global (#2957)	2024-01-01 10:59:56 -08:00
test_lazybuffer.py	scheduleitem is not Tuple [run_process_replay] (#5425)	2024-07-12 15:13:19 -07:00
test_lazyop.py	scheduleitem is not Tuple [run_process_replay] (#5425)	2024-07-12 15:13:19 -07:00
test_linearizer.py	folding without UNMUL (#5628)	2024-07-21 20:14:44 -07:00
test_linearizer_dumb.py	tests if the linearizer is generating dumb code (#5611)	2024-07-20 20:36:32 -07:00
test_linearizer_failures.py	folding without UNMUL (#5628)	2024-07-21 20:14:44 -07:00
test_linearizer_overflows.py	lowerer is kernel [run_process_replay] (#5437)	2024-07-12 18:50:55 -07:00
test_masked_st.py	multitensor start (#2676)	2023-12-07 17:07:05 -08:00
test_method_cache.py	simple LoadOps.ASSIGN (#3745)	2024-03-14 20:44:34 -07:00
test_multitensor.py	smaller multitensor resnet test (#5450)	2024-07-13 07:31:28 -07:00
test_net_speed.py	nv mockgpu (#4600)	2024-05-15 23:46:08 +03:00
test_nn.py	MetaOps.KERNEL (#5543)	2024-07-17 19:41:23 -07:00
test_ops.py	folding without UNMUL (#5628)	2024-07-21 20:14:44 -07:00
test_optim.py	improve test_dropout_on_shard (#4912)	2024-06-11 11:36:02 -04:00
test_pattern_matcher.py	folding without UNMUL (#5628)	2024-07-21 20:14:44 -07:00
test_pickle.py	test pickle variable (#5150)	2024-06-25 19:49:21 -07:00
test_randomness.py	jit sampling functionn in test_randomness.test_multinomial (#5034)	2024-06-18 14:21:05 -04:00
test_sample.py	enable test_sample for all backend (#2593)	2023-12-03 17:20:27 -05:00
test_schedule.py	test: put conv in one reduce (#4441)	2024-07-22 12:16:13 +03:00
test_search.py	BEAM bugfix, kernels dedup now (#5617)	2024-07-20 19:43:50 -07:00
test_setitem.py	setitem in-place operator tests (#4577)	2024-05-14 01:28:02 -04:00
test_specific_conv.py	nv mockgpu (#4600)	2024-05-15 23:46:08 +03:00
test_speed_v_torch.py	move colorize_float to helpers.py (#5490)	2024-07-15 11:29:03 -07:00
test_subbuffer.py	support view in nv/amd (#4812)	2024-06-03 22:11:52 +03:00
test_symbolic_jit.py	sort vars in jit when building expected input args (#4990)	2024-06-16 15:55:51 -04:00
test_symbolic_ops.py	symbolic Tensor.var (#4843)	2024-06-05 12:55:54 -04:00
test_symbolic_shapetracker.py	support symbolic reshape with non-contiguous (#4844)	2024-06-05 16:01:19 -04:00
test_tensor.py	use is to compare types (#5476)	2024-07-14 14:26:41 -04:00
test_tensor_data.py	BEAM_COMPARE=2 validates the correctness of BEAM kernels (#5458)	2024-07-13 13:53:43 -07:00
test_tensor_variable.py	Should this symbolic test fail? (#4501)	2024-06-18 15:21:26 -04:00
test_to_numpy.py	Apply ruff linting rules to tests (#2473)	2023-11-27 21:24:06 -08:00
test_transcendental.py	relax fuzz transend math threshold a bit (#5442)	2024-07-13 03:31:21 -04:00
test_uop_graph.py	move UPat and PatternMatcher from uopgraph.py to uops.py (#5597)	2024-07-19 19:28:24 -04:00
test_uops.py	pretty print lazy op per default (#5505)	2024-07-18 09:34:08 -07:00
test_uops_stats.py	careful memory counting (with tests to specify behavior) (#5587)	2024-07-19 11:37:34 -07:00
test_verify_lazyop.py	pretty print lazy op per default (#5505)	2024-07-18 09:34:08 -07:00
test_winograd.py	MetaOps.KERNEL (#5543)	2024-07-17 19:41:23 -07:00
test_zero_copy.py	remove numpy from device (#3123)	2024-01-14 19:36:05 -08:00