Francis LamandGitHub bbb0ad4800 wmma: widen TC usage in search by using PADTO on TC axes when possible (#4216) ...

* wmma: widen TC usage in search by using PADTO on TC axes when possible

* test: start tests for the new padding TC behavior

* search: upgrade padded TC search to TC_OPT >= 2

* test: add behavior and correctness test for padded TC

added optional argument to apply_tensor_core to set TC_OPT level

* linearizer: add tests for the PADTO behvaior and docs

2024-04-22 16:50:31 -04:00

..

codegen

wmma: widen TC usage in search by using PADTO on TC axes when possible (#4216)

2024-04-22 16:50:31 -04:00

engine

SAVE_SCHEDULE as contextvar (#4230)

2024-04-19 18:51:57 -04:00

features

wmma: widen TC usage in search by using PADTO on TC axes when possible (#4216)

2024-04-22 16:50:31 -04:00

nn

datasets isn't a feature + filter docstrings (#4228)

2024-04-19 16:16:10 +04:00

renderer

Update ssa input order and annotate types in cstyle and assembly (#4117)

2024-04-09 13:10:29 -04:00

runtime

nv driver (#4044)

2024-04-22 19:50:20 +04:00

shape

clean up update stats (#4226)

2024-04-19 15:41:30 +04:00

__init__.py

spend 5 lines to bring mnist into the repo (#4122)

2024-04-09 19:24:57 -07:00

buffer.py

new memory scheduler with explicit refcounts (#4198)

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

device.py

clean up update stats (#4226)

2024-04-19 15:41:30 +04:00

dtype.py

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

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

function.py

New docs are in mkdocs (#4178)

2024-04-16 10:59:51 +04:00

helpers.py

SAVE_SCHEDULE as contextvar (#4230)

2024-04-19 18:51:57 -04:00

lazy.py

new memory scheduler with explicit refcounts (#4198)

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

ops.py

optionally use a copy kernel instead of SDMA (#4116)

2024-04-12 23:10:41 -07:00

tensor.py

some docstrings (#4201)

2024-04-21 16:34:08 +04:00