Allow AMX instructions with K dimension larger than 4 bytes - #6582

Merged

abadams merged 23 commits into

halide:mainfrom

frengels:tile_matmul_k

Aug 1, 2022

Commits on Jul 27, 2022

recognize the patterns used for the RHS matrix
frengels
committed
c32b424
make 1d tile matcher more robust
frengels
committed
43931e9
put getting rhs tile's index into a separate func
frengels
committed
f4a153a
expand the tests used in correctness check
frengels
committed
38a8e96
add exclamation mark
frengels
committed
cbdc68f
remove unused vars
frengels
committed
0e3998e
run format and tidy
frengels
committed
464c240
check for null before using IR in the next step
frengels
committed
c174b98
check if the broadcast was found
frengels
committed
f9086fa
llvm below 13 is no longer supported
frengels
committed
92426c8
replace single pattern with commutative permutations
frengels
committed
fba5a2e
check if the stride is an `IntImm`, otherwise reject pattern
frengels
committed
3764d49
apply clang-format-13
frengels
committed
bd5cd6b
rename wild_i32 -> v2
frengels
committed
ddd3a0e
check if v1 could be the stride value
frengels
committed
db12ffc
add more detail to a receiving a bad type
frengels
committed
b56aad1
added short explanation of the right-hand matrix layout
frengels
committed
7be1b5b
added explanation for where the 4 comes from
frengels
committed
81bacbb
provide further documentation as to the layout of AMX
frengels
committed
a80fc76
Merge branch 'main' into pr/6582
steven-johnson
committed
1366699
Merge branch 'main' into pr/6582
steven-johnson
committed
fc89f67

Commits on Jul 29, 2022

add comments for expected patterns to get_3d_rhs_tile_index
frengels
committed
314656c

Commits on Aug 1, 2022

Document the matched pattern
frengels
committed
9b30679