[Mlir-commits] [mlir] [MLIR][test] Check for ml_dtypes before running tests (PR #123061)
Maksim Levental
llvmlistbot at llvm.org
Thu Jan 16 08:03:39 PST 2025
makslevental wrote:
> The expected `# CHECK-LABEL: TEST: testBF16Memref` and `# CHECK: [0.5]` will not be found.
>
> What options do we have here?
whoops should've thought of that - in this case the solution is simple: just change those checks for printed values into `assert`s for those same values:
```python
x = ranked_memref_to_numpy(arg2_memref_ptr[0])
assert len(x) == 1 and assert(x[0]) == 0.5
```
https://github.com/llvm/llvm-project/pull/123061
More information about the Mlir-commits
mailing list