madt2709andGitHub 4bb98d8882 Fix track_running_stats in batchnorm (#6200) ...

* Fix track_running_stats in batchnorm

* Fix linter

* Update test_fold_conv_batchnorm_notrain to keep allowed at 1

* Add test_fold_conv_batchnorm_notrain_no_running_stats

* Save 1 line

2024-08-20 14:01:22 -07:00

..

codegen

fix indexing out of bounds (#6208)

2024-08-20 11:34:56 -07:00

engine

hotfix: more lazyop rename to uop [run_process_replay] (#6157)

2024-08-18 17:28:44 +03:00

nn

Fix track_running_stats in batchnorm (#6200)

2024-08-20 14:01:22 -07:00

renderer

Refactor amd kernel prefix (#6205)

2024-08-20 10:37:36 -07:00

runtime

_gpu_alloc -> allocator.alloc (#6189)

2024-08-19 23:34:22 +03:00

shape

re:6125 switch real_size to use uops [run_process_replay] (#6138)

2024-08-19 13:20:24 -04:00

__init__.py

add mnist tutorial

2024-04-25 16:08:32 +08:00

device.py

RUF018 assignment-in-assert [run_process_replay] (#6172)

2024-08-19 00:34:52 -04:00

dtype.py

threefry touchup [run_process_replay] (#6169)

2024-08-18 23:01:24 -04:00

function.py

remove UnaryOps.NEG from function.py (#6187)

2024-08-19 17:39:15 -04:00

helpers.py

Support gunzip in fetch (#6176)

2024-08-19 12:04:40 -07:00

lazy.py

remove UnaryOps.NEG from lazy.py (#6193)

2024-08-19 18:41:28 -04:00

multi.py

MLB support reshape for uneven shards (#5804)

2024-08-01 02:36:03 -07:00

ops.py

type verify intermediate UOps [run_process_replay] (#6140)

2024-08-19 20:59:01 +03:00

py.typed

add a single py.typed (#6083)

2024-08-14 17:31:46 -07:00

tensor.py

remove UnaryOps.NEG from function.py (#6187)

2024-08-19 17:39:15 -04:00