Skip to content

[mlir][Aarch64] Add lowering support from unsigned vector.contract to i8mm #90412

Open
@dcaballe

Description

@dcaballe

LowerContractionToSMMLAPattern.cpp added support to lower a vector.contract to i8mm's smmla instructions. However, the existing code can only handle signed i8,i8,i32 matrix multiplication cases. We should generalize (and rename) the pass to also handle unsigned i8,i8,i32 cases by adding matching support for extui and generating ummla instructions instead.

#Gemma

Metadata

Metadata

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions