geohotstan a684d72e55 add ceil_mode for avg_pool and max_pool (#7579) ... * wip pool * check CI for remove alternative implementation * Revert "check CI for remove alternative implementation" This reverts commit 7b1bb900e5. * fix test * tests tests tests * slap a resolve on it * fix comment * a little simpler pool * check CI for removal again * Revert "check CI for removal again" This reverts commit be798b7857. * small * update * some ez tests * english * clean up code * fix ruff * how did I +25 lines? * small clean ups * moar clean ups * try test_avgpool2d_failure2 in CI * final clean up * exclude bug fix * avg underscore pool * no more edge case stuff * add better comments for explanation * add test cases for decreasing end padding * address feedback * improve test coverage * tiny more polish as we wait for lines :D * more readable code ordering * add to documentation * oops * set to False instead --------- Co-authored-by: chenyu <chenyu@fastmail.com>		2024-12-06 08:34:14 -05:00
..
external	Revert "combine get inputs and type_parse function in onnx (#8069)" (#8079)	2024-12-06 08:04:21 -05:00
imported	no type-tracker getitem refactor (#6917)	2024-11-29 10:18:02 -05:00
models	update kernel counts in test_real_world (#7960)	2024-11-29 11:14:54 -05:00
testextra	u32 to f16 in tinygrad (#8074)	2024-12-06 12:00:13 +01:00
unit	improve idiv _min_max (#8066)	2024-12-05 23:02:16 -05:00
web	Bring back WebGPU (#7063)	2024-11-26 12:26:40 +08:00
__init__.py	All devices are equal! (#196)	2020-12-15 23:44:08 -08:00
Dockerfile	new cloud is cloudy [pr] (#7631)	2024-11-11 20:18:04 +08:00
helpers.py	move is_dtype_supported to device [pr] (#7575)	2024-11-07 20:38:03 +08:00
test_arange.py	Fix WebGPU atomic store (#7954)	2024-11-29 19:31:25 +08:00
test_assign.py	Add replicate mode to Tensor.pad (#7608)	2024-11-18 10:55:38 -05:00
test_compile_failures.py	enable a few tests that work now (#7721)	2024-11-15 14:30:52 -05:00
test_const_folding.py	reduce folding simple tests [pr] (#8040)	2024-12-05 12:22:45 +08:00
test_conv.py	db in wal mode (#5388)	2024-07-12 20:43:36 -07:00
test_conv_shapetracker.py	clean up a few parents -> toposort [pr] (#7984)	2024-12-02 15:59:31 +08:00
test_copy_speed.py	node cleanup + local metal test speed [pr] (#6880)	2024-10-04 18:14:23 +08:00
test_device_speed.py	remove iter from uopgraph (#6110)	2024-08-16 15:58:29 -07:00
test_dtype.py	Remove WebGL (#8012)	2024-12-03 16:02:53 +01:00
test_dtype_alu.py	Run WebGPU tests on ubuntu (#8033)	2024-12-04 12:42:04 +01:00
test_fusion_op.py	Ops.VALID cleanup, move recursive tests [pr] (#7713)	2024-11-15 20:22:46 +08:00
test_fuzz_shape_ops.py	faster ci [pr] (#7348)	2024-10-29 14:01:44 +08:00
test_gc.py	small changes from block linearizer [pr] (#7888)	2024-11-25 15:27:04 +08:00
test_graph.py	view doesn't have buffer, fix the tests [pr] (#7841)	2024-11-22 20:41:55 +08:00
test_hcq.py	hcq replace update with sint (#7899)	2024-11-29 20:08:13 +03:00
test_image_dtype.py	don't mutate the uop/lazybuffer, just the Buffer [pr] (#8000)	2024-12-03 19:03:51 +08:00
test_jit.py	qcom has not transfer (#8075)	2024-12-06 14:45:01 +03:00
test_kernel_cache.py	test_kernel_cache_in_action: fix test (#7792)	2024-11-19 13:34:56 -05:00
test_lazybuffer.py	assert view degrade to const tests post scheduler graph_rewrite [pr] (#7822)	2024-11-21 19:00:41 +08:00
test_linearizer.py	no bool in range [pr] (#7988)	2024-12-02 19:05:16 +08:00
test_linearizer_dumb.py	replace all sparents with toposort (#7983)	2024-12-02 15:00:30 +08:00
test_linearizer_failures.py	replace all sparents with toposort (#7983)	2024-12-02 15:00:30 +08:00
test_linearizer_overflows.py	Remove UnaryOps, BinaryOps, TernaryOps, MetaOps [pr] (#7725)	2024-11-16 20:56:56 +08:00
test_masked_st.py	multitensor start (#2676)	2023-12-07 17:07:05 -08:00
test_metal.py	Use MTLCompiler directly (v2) (#7920)	2024-12-04 16:36:48 +08:00
test_method_cache.py	simple LoadOps.ASSIGN (#3745)	2024-03-14 20:44:34 -07:00
test_multitensor.py	fix Tensor(MultiLazyBuffer) with different dtype should fail (#7757)	2024-11-17 21:05:45 -05:00
test_net_speed.py	nv mockgpu (#4600)	2024-05-15 23:46:08 +03:00
test_nn.py	Raise error when model architecture does not match state dict (#7772)	2024-11-20 00:11:54 +08:00
test_ocl.py	Fix GPT2 with OpenCL backend. (#6821)	2024-10-01 16:57:22 +08:00
test_ops.py	add ceil_mode for avg_pool and max_pool (#7579)	2024-12-06 08:34:14 -05:00
test_optim.py	move is_dtype_supported to device [pr] (#7575)	2024-11-07 20:38:03 +08:00
test_pickle.py	Revert "s/UPat/Pat (#7506)" [pr] (#7517)	2024-11-03 16:33:02 -05:00
test_profiler.py	_prg to match _buf [pr] (#7816)	2024-11-21 12:44:48 +08:00
test_randomness.py	add tiny test for randomness + remove ulong buffers (#7648)	2024-11-12 12:45:52 +08:00
test_rearrange_einops.py	Missing features from rearrange (#6184)	2024-08-19 11:19:07 -07:00
test_renderer_failures.py	BufferSpec and ProgramSpec [pr] (#7814)	2024-11-21 12:18:05 +08:00
test_sample.py	UOp.variable (#7010)	2024-10-12 18:20:44 +08:00
test_schedule.py	assert const folding return shape matches [pr] (#8006)	2024-12-03 19:31:06 +08:00
test_search.py	clean up a few parents -> toposort [pr] (#7984)	2024-12-02 15:59:31 +08:00
test_setitem.py	fix advanced setitem with 1 in shape (#7797)	2024-11-19 20:04:59 -05:00
test_specific_conv.py	move is_dtype_supported to device [pr] (#7575)	2024-11-07 20:38:03 +08:00
test_speed_v_torch.py	fix HALF=1 in test_speed_v_torch (#7642)	2024-11-11 14:29:37 -05:00
test_subbuffer.py	move lazy to engine [pr] (#6886)	2024-10-04 23:19:26 +08:00
test_symbolic_jit.py	UOp.variable (#7010)	2024-10-12 18:20:44 +08:00
test_symbolic_ops.py	UOp.variable (#7010)	2024-10-12 18:20:44 +08:00
test_symbolic_shapetracker.py	fix view add with symbolic shape (#7569)	2024-11-06 11:39:20 -05:00
test_tensor.py	clean up reshape size check [pr] (#8067)	2024-12-06 07:51:19 -05:00
test_tensor_data.py	[BUGFIX] Tensor([]).data() (#7884)	2024-11-24 16:42:57 -05:00
test_tensor_variable.py	combine pad2d with pad (#7677)	2024-11-14 17:56:02 +08:00
test_tiny.py	process replayable ops.py changes from delete_lazy [pr] (#7994)	2024-12-02 19:38:31 +08:00
test_to_numpy.py	Apply ruff linting rules to tests (#2473)	2023-11-27 21:24:06 -08:00
test_transcendental.py	move is_dtype_supported to device [pr] (#7575)	2024-11-07 20:38:03 +08:00
test_uop_graph.py	update uop and tests to not use lt/gt/le/ge [pr] (#8023)	2024-12-03 21:02:52 -05:00
test_uops.py	make device on uop optional [pr] (#8034)	2024-12-04 20:18:00 +08:00
test_uops_stats.py	update tests (#7554)	2024-11-05 11:35:15 -05:00
test_viz.py	viz with bottom_up=True (#7894)	2024-11-25 17:56:48 +08:00
test_winograd.py	clean up a few parents -> toposort [pr] (#7984)	2024-12-02 15:59:31 +08:00
test_zero_copy.py	remove numpy from device (#3123)	2024-01-14 19:36:05 -08:00