Francis LamandGitHub c12bcabb07 search: fix actions space checks to ignore TC axis and amt (#4360) ...

* search: fix actions space checks to ignore TC axis and amt

* add test for number of actions in get_linearizer_actions

2024-04-30 14:02:22 -04:00

..

codegen

wmma: add reduce axis choice to TC action space (#4328)

2024-04-29 19:15:39 -04:00

engine

save schedule graph pre validation (#4317)

2024-04-27 12:06:15 +03:00

features

search: fix actions space checks to ignore TC axis and amt (#4360)

2024-04-30 14:02:22 -04:00

nn

update docs (#4356)

2024-04-30 16:51:42 +09:00

renderer

Ptx beam fix (#4296)

2024-04-25 15:39:39 -04:00

runtime

refactor ops_gpu ctypes (#4331)

2024-04-30 01:33:34 +08:00

shape

Fix in _reshape_mask (#4332)

2024-04-28 11:57:39 -04:00

__init__.py

add mnist tutorial

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

buffer.py

new memory scheduler with explicit refcounts (#4198)

2024-04-17 08:46:47 +04:00

device.py

more docs (#4271)

2024-04-24 10:52:42 +08:00

dtype.py

rename Scalar to ConstType and cast_scalar to as_const (#3946)

2024-03-26 22:39:58 -04:00

function.py

fix function.py sum backward without downcast_half (#4353)

2024-04-29 17:53:02 -04:00

helpers.py

search: add additional 7 factors to the action space (#4256)

2024-04-22 19:14:23 -04:00

lazy.py

fix dtype of const folded sum (#4349)

2024-04-29 11:40:45 -04:00

ops.py

update docs (#4356)

2024-04-30 16:51:42 +09:00

tensor.py

just get dtype from kwargs (#4355)

2024-04-30 16:26:14 +09:00