oplavsic 715a589ce3 [FA fwd D=128] Reduce LDS usage in epilogue (#340) ...

* rebase onto improve_fwd_fa

* Fixed a leftover from rebase

* rebase onto improve_fa_fwd

* Reduce tuning space

* Disable bwd with D=128

* Add test for d=128

* Fix an issue with get_best_config when there is only one config

* Added better configs for d=128

* Fix typos

---------

Co-authored-by: Lixun Zhang <lixun.zhang@amd.com>

2023-10-25 12:10:34 -05:00

..

backends

[FRONTEND] drop the GIL around more CUDA ops (#2173)

2023-08-24 20:31:38 -07:00

__init__.py

[RUNTIME] refactor driver (#1515)

2023-04-12 23:50:44 -07:00

autotuner.py

[FA fwd D=128] Reduce LDS usage in epilogue (#340)

2023-10-25 12:10:34 -05:00

cache.py

[RUNTIME] Fix cache dir (#2196)

2023-08-29 21:07:16 -04:00

driver.py

Initial code merge of Hopper support (#2036)

2023-08-07 09:53:04 +08:00

errors.py

[FRONTEND] cleaned up language; added frontend function for globaltimer special register (#1525)

2023-04-14 15:29:27 -07:00

jit.py

[MFMA] Switch between MFMA types (#352)

2023-10-18 16:57:34 +02:00