Linalg->Vector dialect pass and pipeline changes

plaidml / tpp-mlir

TPP experimentation on MLIR for linear algebra

https://arxiv.org/abs/2404.15204

Other

111 stars 29 forks source link

Linalg->Vector dialect pass and pipeline changes #958

Closed KavithaTipturMadhu closed 4 weeks ago

KavithaTipturMadhu commented 1 month ago

Working on test cases, will update shortly, only small tensors lower right now as vectorization lowers them to extract-insert operations, need to investigate what happens at lowering. Also, options need to be added for vector length and other command line arguments supported upstream. Working on those, will update the PR asap.

KavithaTipturMadhu commented 1 month ago

One set of tests added, pr is good for review, will add e2e tests for correctness and cli options next.

KavithaTipturMadhu commented 1 month ago

Added bf16 end to end test case. The pr is good to go @adam-smnk @rolfmorel. Thanks @rengolin for the pointers on fpcmp.