diff options
Diffstat (limited to 'mlir/python/CMakeLists.txt')
| -rw-r--r-- | mlir/python/CMakeLists.txt | 9 |
1 files changed, 9 insertions, 0 deletions
diff --git a/mlir/python/CMakeLists.txt b/mlir/python/CMakeLists.txt index 84ff08a7631d..e79b51ad72e2 100644 --- a/mlir/python/CMakeLists.txt +++ b/mlir/python/CMakeLists.txt @@ -349,6 +349,15 @@ declare_mlir_dialect_python_bindings( declare_mlir_dialect_python_bindings( ADD_TO_PARENT MLIRPythonSources.Dialects ROOT_DIR "${CMAKE_CURRENT_SOURCE_DIR}/mlir" + TD_FILE dialects/ShardOps.td + SOURCES + dialects/shard.py + DIALECT_NAME shard + GEN_ENUM_BINDINGS) + +declare_mlir_dialect_python_bindings( + ADD_TO_PARENT MLIRPythonSources.Dialects + ROOT_DIR "${CMAKE_CURRENT_SOURCE_DIR}/mlir" TD_FILE dialects/MLProgramOps.td SOURCES dialects/ml_program.py |