[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