tinygrad/tinygrad
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
..
codegen Cleaner rule for mul/idiv by power of two [pr] (#8076) 2024-12-06 08:02:24 -05:00
engine fix bug in jit prune with copy [pr] (#8073) 2024-12-06 18:38:23 +08:00
nn refactor safe_load (#8035) 2024-12-06 12:08:21 +08:00
renderer update uop and tests to not use lt/gt/le/ge [pr] (#8023) 2024-12-03 21:02:52 -05:00
runtime qcom has not transfer (#8075) 2024-12-06 14:45:01 +03:00
shape clean up reshape size check [pr] (#8067) 2024-12-06 07:51:19 -05:00
viz replace all sparents with toposort (#7983) 2024-12-02 15:00:30 +08:00
__init__.py UOp.variable (#7010) 2024-10-12 18:20:44 +08:00
device.py PYTHON device is okay to use everywhere [pr] (#7981) 2024-12-02 12:34:42 +08:00
dtype.py mypy==1.13.0 (#7990) 2024-12-06 12:09:14 +08:00
function.py update function to not use gt/lt [pr] (#8025) 2024-12-03 22:39:06 -05:00
helpers.py small helpers cleanups (#7977) 2024-12-01 21:50:47 -05:00
multi.py least controversial (#7863) 2024-11-23 21:23:30 +08:00
ops.py improve idiv _min_max (#8066) 2024-12-05 23:02:16 -05:00
py.typed add a single py.typed (#6083) 2024-08-14 17:31:46 -07:00
tensor.py add ceil_mode for avg_pool and max_pool (#7579) 2024-12-06 08:34:14 -05:00