You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
I'm interested in using the torch-mlir python bindings in a project that uses Bazel as its build system. Based on the developer docs, it seems like the Bazel configs for torch-mlir are mostly focused on supporting torch-mlir-opt, and indeed I don't see any CAPI or Python binding rules in the main Bazel build file.
I think what's needed to do this is:
The CAPI library (from torch-mlir-c/Dialects.h). This can be built with mlir_c_api_cc_library from @llvm-project//mlir:build_defs.bzl.
The pybind11 extension library (from python/TorchMLIRModule.cpp).
A tablegen rule to generate the Python wrapper code for ops (_torch_ops_gen.py).
(Maybe) A convenience py_library that bundles the above and any MLIR Python dependencies.
There are examples of these in the MLIR project's Bazel file.
Is there interest in this from the maintainers? I'd be willing to contribute these changes if so.
The text was updated successfully, but these errors were encountered:
I'm interested in using the torch-mlir python bindings in a project that uses Bazel as its build system. Based on the developer docs, it seems like the Bazel configs for torch-mlir are mostly focused on supporting
torch-mlir-opt
, and indeed I don't see any CAPI or Python binding rules in the main Bazel build file.I think what's needed to do this is:
torch-mlir-c/Dialects.h
). This can be built withmlir_c_api_cc_library
from@llvm-project//mlir:build_defs.bzl
.python/TorchMLIRModule.cpp
)._torch_ops_gen.py
).py_library
that bundles the above and any MLIR Python dependencies.There are examples of these in the MLIR project's Bazel file.
Is there interest in this from the maintainers? I'd be willing to contribute these changes if so.
The text was updated successfully, but these errors were encountered: