[all-commits] [llvm/llvm-project] 89801c: [MLIR][Python] add ctype python binding support fo...

Bimo via All-commits all-commits at lists.llvm.org
Wed May 29 22:02:02 PDT 2024


  Branch: refs/heads/main
  Home:   https://github.com/llvm/llvm-project
  Commit: 89801c74c3e25f5a1eaa3999863be398f6a82abb
      https://github.com/llvm/llvm-project/commit/89801c74c3e25f5a1eaa3999863be398f6a82abb
  Author: Bimo <rui.xu at intel.com>
  Date:   2024-05-29 (Wed, 29 May 2024)

  Changed paths:
    M mlir/python/mlir/runtime/np_to_memref.py
    M mlir/python/requirements.txt
    M mlir/test/python/execution_engine.py

  Log Message:
  -----------
  [MLIR][Python] add ctype python binding support for bf16 (#92489)

Since bf16 is supported by mlir, similar to
complex128/complex64/float16, we need an implementation of bf16 ctype in
Python binding. Furthermore, to resolve the absence of bf16 support in
NumPy, a third-party package [ml_dtypes
](https://github.com/jax-ml/ml_dtypes) is introduced to add bf16
extension, and the same approach was used in `torch-mlir` project.

See motivation and discussion in:
https://discourse.llvm.org/t/how-to-run-executionengine-with-bf16-dtype-in-mlir-python-bindings/79025



To unsubscribe from these emails, change your notification settings at https://github.com/llvm/llvm-project/settings/notifications


More information about the All-commits mailing list