Skip to main content

Browse the archive

https://github.com/halide/Halide
21 March 2024, 02:33:35 UTC
Revision 8871404403554d7f001c81ee10fa35cc42fe0cca authored by Frederik on 01 August 2022, 23:23:28 UTC, committed by GitHub on 01 August 2022, 23:23:28 UTC 
Allow AMX instructions with K dimension larger than 4 bytes (#6582)
 
* recognize the patterns used for the RHS matrix

* make 1d tile matcher more robust

* put getting rhs tile's index into a separate func

* expand the tests used in correctness check

* add exclamation mark

* remove unused vars

* run format and tidy

* check for null before using IR in the next step

* check if the broadcast was found

* llvm below 13 is no longer supported

* replace single pattern with commutative permutations

* check if the stride is an `IntImm`, otherwise reject pattern

* apply clang-format-13

* rename wild_i32 -> v2

* check if v1 could be the stride value

* add more detail to a receiving a bad type

* added short explanation of the right-hand matrix layout

* added explanation for where the 4 comes from

* provide further documentation as to the layout of AMX

* add comments for expected patterns to get_3d_rhs_tile_index

* Document the matched pattern

Co-authored-by: Steven Johnson <srj@google.com>
1 parent 703a738
History
Tip revision: 8871404403554d7f001c81ee10fa35cc42fe0cca authored by Frederik on 01 August 2022, 23:23:28 UTC
Allow AMX instructions with K dimension larger than 4 bytes (#6582)
Tip revision: 8871404
File Mode Size
.github
apps
cmake
dependencies
doc
packaging
python_bindings
src
test
tools
tutorial
util
.clang-format -rw-r--r-- 1.4 KB
.clang-format-ignore -rw-r--r-- 286 bytes
.clang-tidy -rw-r--r-- 2.2 KB
.gitattributes -rw-r--r-- 342 bytes
.gitignore -rw-r--r-- 4.9 KB
.gitmodules -rw-r--r-- 0 bytes
CMakeLists.txt -rw-r--r-- 5.7 KB
CMakePresets.json -rw-r--r-- 5.0 KB
CODE_OF_CONDUCT.md -rw-r--r-- 3.5 KB
LICENSE.txt -rw-r--r-- 3.2 KB
Makefile -rw-r--r-- 102.5 KB
README.md -rw-r--r-- 16.4 KB
README_cmake.md -rw-r--r-- 72.4 KB
README_rungen.md -rw-r--r-- 12.1 KB
README_webassembly.md -rw-r--r-- 10.5 KB
run-clang-format.sh -rwxr-xr-x 1.4 KB
run-clang-tidy.sh -rwxr-xr-x 3.1 KB

README.md

back to top