search

nod-ai / iree-amd-aie

IREE plugin repository for the AMD AIE accelerator
Apache License 2.0
45 stars 23 forks source link

Support this accumulating matmul #449

Open newling opened 1 week ago

newling commented 1 week ago

Example reported on discord:

func.func @matmul_small(%lhs : tensor<32x16xi32>,
    %rhs : tensor<16x32xi32>, 
    %acc : tensor<32x32xi32>) -> tensor<32x32xi32> {
  %2 = linalg.matmul ins(%lhs, %rhs : tensor<32x16xi32>, tensor<16x32xi32>)
     outs(%acc: tensor<32x32xi32>) -> tensor<32x32xi32>
  return %2 : tensor<32x32xi32>
}

Results in stack trace without any useful information