[Codegen][GPU] Make operand promotion pattern work with generics

iree-org / iree

A retargetable MLIR-based machine learning compiler and runtime toolkit.

http://iree.dev/

Apache License 2.0

2.47k stars 548 forks source link

[Codegen][GPU] Make operand promotion pattern work with generics #17650

Closed qedawkins closed 2 weeks ago

qedawkins commented 2 weeks ago

The pattern was previously using the isMatmulOrBatchMatmul helper that only looked for named ops. Change the logic to use inferred contraction dims and look at the static bounds of the op to filter out matvec cases.