[Mlir-commits] [mlir] Support float8_e3m4 and float8_e4m3 in np_to_memref (PR #186453)

[LLVM Continuous Integration]

llvm-ci wrote:

LLVM Buildbot has detected a new failure on builder `mlir-nvidia` running on `mlir-nvidia` while building `mlir` at step 7 "test-build-check-mlir-build-only-check-mlir".

Full details are available at: https://lab.llvm.org/buildbot/#/builders/138/builds/27095

<details>
<summary>Here is the relevant piece of the build log for the reference</summary>

```
Step 7 (test-build-check-mlir-build-only-check-mlir) failure: test (failure)
******************** TEST 'MLIR :: Examples/NVGPU/Ch1.py' FAILED ********************
Exit Code: 2

Command Output (stdout):
--
# RUN: at line 1
env SUPPORT_LIB=/vol/worker/mlir-nvidia/mlir-nvidia/llvm.obj/lib/libmlir_cuda_runtime.so  sh -c 'if [[ "%mlir_run_cuda_sm90_tests" == "1" ]];  then /usr/bin/python3.10 /vol/worker/mlir-nvidia/mlir-nvidia/llvm.src/mlir/test/Examples/NVGPU/Ch1.py | /vol/worker/mlir-nvidia/mlir-nvidia/llvm.obj/bin/FileCheck /vol/worker/mlir-nvidia/mlir-nvidia/llvm.src/mlir/test/Examples/NVGPU/Ch1.py;  else export MLIR_NVDSL_PRINT_IR=1;  /usr/bin/python3.10 /vol/worker/mlir-nvidia/mlir-nvidia/llvm.src/mlir/test/Examples/NVGPU/Ch1.py | /vol/worker/mlir-nvidia/mlir-nvidia/llvm.obj/bin/FileCheck /vol/worker/mlir-nvidia/mlir-nvidia/llvm.src/mlir/test/Examples/NVGPU/Ch1.py --check-prefix=DUMPIR; fi'
# executed command: env SUPPORT_LIB=/vol/worker/mlir-nvidia/mlir-nvidia/llvm.obj/lib/libmlir_cuda_runtime.so sh -c 'if [[ "%mlir_run_cuda_sm90_tests" == "1" ]];  then /usr/bin/python3.10 /vol/worker/mlir-nvidia/mlir-nvidia/llvm.src/mlir/test/Examples/NVGPU/Ch1.py | /vol/worker/mlir-nvidia/mlir-nvidia/llvm.obj/bin/FileCheck /vol/worker/mlir-nvidia/mlir-nvidia/llvm.src/mlir/test/Examples/NVGPU/Ch1.py;  else export MLIR_NVDSL_PRINT_IR=1;  /usr/bin/python3.10 /vol/worker/mlir-nvidia/mlir-nvidia/llvm.src/mlir/test/Examples/NVGPU/Ch1.py | /vol/worker/mlir-nvidia/mlir-nvidia/llvm.obj/bin/FileCheck /vol/worker/mlir-nvidia/mlir-nvidia/llvm.src/mlir/test/Examples/NVGPU/Ch1.py --check-prefix=DUMPIR; fi'
# .---command stderr------------
# | sh: 1: [[: not found
# | Traceback (most recent call last):
# |   File "/vol/worker/mlir-nvidia/mlir-nvidia/llvm.src/mlir/test/Examples/NVGPU/Ch1.py", line 64, in <module>
# |     saxpy(x, y, alpha)
# |   File "/vol/worker/mlir-nvidia/mlir-nvidia/llvm.src/mlir/test/Examples/NVGPU/tools/nvdsl.py", line 410, in wrapper
# |     types.append(get_mlir_ty(arg))
# |   File "/vol/worker/mlir-nvidia/mlir-nvidia/llvm.src/mlir/test/Examples/NVGPU/tools/nvdsl.py", line 298, in get_mlir_ty
# |     descriptor = rt.get_ranked_memref_descriptor(arg)
# |   File "/vol/worker/mlir-nvidia/mlir-nvidia/llvm.obj/tools/mlir/python_packages/mlir_core/mlir/runtime/np_to_memref.py", line 138, in get_ranked_memref_descriptor
# |     ctp = as_ctype(nparray.dtype)
# |   File "/vol/worker/mlir-nvidia/mlir-nvidia/llvm.obj/tools/mlir/python_packages/mlir_core/mlir/runtime/np_to_memref.py", line 72, in as_ctype
# |     if ml_dtypes is not None and dtp == ml_dtypes.float8_e3m4:
# | AttributeError: module 'ml_dtypes' has no attribute 'float8_e3m4'. Did you mean: 'float8_e5m2'?
# | FileCheck error: '<stdin>' is empty.
# | FileCheck command line:  /vol/worker/mlir-nvidia/mlir-nvidia/llvm.obj/bin/FileCheck /vol/worker/mlir-nvidia/mlir-nvidia/llvm.src/mlir/test/Examples/NVGPU/Ch1.py --check-prefix=DUMPIR
# `-----------------------------
# error: command failed with exit status: 2

--

********************


```

</details>

https://github.com/llvm/llvm-project/pull/186453