TOSA dialect type inference error

[gmmyung]

Hi,

I am currently working on a PoC of a AOT compiled ML framework in Rust using eerie (Rust binding to IREE).

I am fairly new to MLIR, and struggling from the following error:

pub fn new_tensor<'a, T>(&'a mut self, shape: &'a [u64]) -> impl Tensor + 'a {
      let location = Location::unknown(&self.melior_context);
      let values: Vec<_> = vec![1.0f32, 2.0f32, 3.0f32, 4.0f32]
          .iter()
          .map(|x| {
              FloatAttribute::new(
                  &self.melior_context,
                  *x as f64,
                  Type::float32(&self.melior_context),
              )
              .into()
          })
          .collect();
      let mlir_type = RankedTensorType::new(shape, Type::float32(&self.melior_context), None);
      let attr =
          DenseElementsAttribute::new(mlir_type.into(), &values).unwrap();
      println!("{:?}", attr);
      let op = tosa::r#const(&self.melior_context, attr.into(), location);
      ConstTensor { op, shape }
 }

warning: `eerietest` (bin "eerietest") generated 3 warnings (run `cargo fix --bin "eerietest"` to apply 2 suggestions)
    Finished dev [unoptimized + debuginfo] target(s) in 1.36s
     Running `target/debug/eerietest`
dense<[[1.000000e+00, 2.000000e+00], [3.000000e+00, 4.000000e+00]]> : tensor<2x2xf32>
error: type inference was requested for the operation tosa.const, but the operation does not support type inference; result types must be specified explicitly
thread 'main' panicked at /Users/gmmyung/.cargo/registry/src/index.crates.io-6f17d22bba15001f/melior-0.13.0/src/dialect/ods.rs:99:1:
valid operation: OperationBuild

What am I doing wrong? Any suggestions will be appreciated!

[raviqqe]

It's probably a bug in Melior's mis-infering availability of type inference or TOSA dialect's configuration.