Request i1 type support in Interpreter

[qingyunqu]

Request description

It failed when I run some case like:

module {
  func.func @forward(%arg0: tensor<4xf32>) -> tensor<4xi1> {
    %0 = stablehlo.convert %arg0 : (tensor<4xf32>) -> tensor<4xi1>
    return %0 : tensor<4xi1>
  }
}

[sdasgup3]

Can you please elaborate on the error message?

Actually, the i1 type is supported in the interpreter. For example,

$ cat test.mlir
module {
  func.func @forward() -> tensor<4xi1> {
     %cst = stablehlo.constant dense<3.0> : tensor<4xf32>
    %0 = stablehlo.convert %cst : (tensor<4xf32>) -> tensor<4xi1>
    return %0 : tensor<4xi1>
  }
}
$ stablehlo-translate --interpret test.mlir
tensor<4xi1> {
  [true, true, true, true]
}

Note: I have modified the above program to specify some input to convert op.

If I run the above mentioned program as is we will get something like (from cs)

error: incorrect number of arguments specified, provided 0 inputs but function expected 1

which means that we have not provided constant inputs equal to the number of program arguments.
Please refer to

stablehlo/examples/c++/ExampleAdd.cpp

Line 91 in 726ac03

auto results = evalModule(*module, {inputValue1, inputValue2}, config);

and

stablehlo/stablehlo/integrations/python/tests/stablehlo.py

Line 283 in 726ac03

actual = np.array(stablehlo.eval_module(m, args)[0])

on how to use the interpreter programatcally using c+ and python APIs resp.

Please let me know if this helps.