Skip to content

Finish linalg.batch_matmul #54

@Balint-R

Description

@Balint-R

This is basically implemented, but I made it throw a NotImplementedError, since it looked like there were a few things missing. Notably, the typeFn variable is not used anywhere, even though other linalg operations use it for casting, and it currently uses tensor.empty which isn't really implemented. Maybe at first, instead of having tensor.empty, we can just make the user pass in the output, like what MLIR does and what we currently do for the other linalg operations. Also should write at least one testcase for batch_matmul.

Metadata

Metadata

Assignees

Labels

featureNew feature or requestsmallNot much effort

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions