[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