mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-18 12:56:07 +00:00
Code Issues Packages Projects Releases Wiki Activity

Remove the ml_program MLIR dialect from jaxlib.

Browse Source 
Jax isn't using this, and in fact our code to build this wasn't including the C++ parts, so it was broken anyway. Remove it until someone actually needs it for something.

PiperOrigin-RevId: 587323808
This commit is contained in:
Peter Hawkins 2023-12-02 09:29:06 -08:00 committed by jax authors
parent 04183707c2
commit 32fb1b4034
5 changed files with 0 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
jax/_src/lib/BUILD
View File

@ -48,7 +48,6 @@ py_library_providing_imports_info(
"//jaxlib/mlir:math_dialect",
"//jaxlib/mlir:memref_dialect",
"//jaxlib/mlir:mhlo_dialect",
"//jaxlib/mlir:ml_program_dialect",
"//jaxlib/mlir:pass_manager",
"//jaxlib/mlir:scf_dialect",
"//jaxlib/mlir:sparse_tensor_dialect",

1
jax/_src/lib/mlir/dialects/__init__.py
View File

@ -20,7 +20,6 @@ import jaxlib.mlir.dialects.math as math
import jaxlib.mlir.dialects.memref as memref
import jaxlib.mlir.dialects.mhlo as mhlo
import jaxlib.mlir.dialects.func as func
import jaxlib.mlir.dialects.ml_program as ml_program
import jaxlib.mlir.dialects.scf as scf
import jaxlib.mlir.dialects.sparse_tensor as sparse_tensor
import jaxlib.mlir.dialects.vector as vector

1
jaxlib/BUILD
View File

@ -62,7 +62,6 @@ py_library_providing_imports_info(
"//jaxlib/mlir:ir",
"//jaxlib/mlir:memref_dialect",
"//jaxlib/mlir:mhlo_dialect",
"//jaxlib/mlir:ml_program_dialect",
"//jaxlib/mlir:pass_manager",
"//jaxlib/mlir:scf_dialect",
"//jaxlib/mlir:sparse_tensor_dialect",

13
jaxlib/mlir/BUILD.bazel
View File

@ -126,19 +126,6 @@ symlink_inputs(
],
)
symlink_inputs(
name = "ml_program_dialect",
rule = py_library,
symlinked_inputs = {"srcs": {"dialects": [
"@llvm-project//mlir/python:MLProgramOpsPyFiles",
]}},
deps = [
":core",
":ir",
":mlir",
],
)
symlink_inputs(
name = "builtin_dialect",
rule = py_library,

2
jaxlib/tools/build_wheel.py
View File

@ -276,7 +276,6 @@ def prepare_wheel(sources_path: pathlib.Path, *, cpu, include_gpu_plugin_extensi
"__main__/jaxlib/mlir/dialects/_math_ops_gen.py",
"__main__/jaxlib/mlir/dialects/_memref_ops_gen.py",
"__main__/jaxlib/mlir/dialects/_mhlo_ops_gen.py",
"__main__/jaxlib/mlir/dialects/_ml_program_ops_gen.py",
"__main__/jaxlib/mlir/dialects/_ods_common.py",
"__main__/jaxlib/mlir/dialects/_scf_ops_gen.py",
"__main__/jaxlib/mlir/dialects/_sparse_tensor_enum_gen.py",
@ -291,7 +290,6 @@ def prepare_wheel(sources_path: pathlib.Path, *, cpu, include_gpu_plugin_extensi
"__main__/jaxlib/mlir/dialects/math.py",
"__main__/jaxlib/mlir/dialects/memref.py",
"__main__/jaxlib/mlir/dialects/mhlo.py",
"__main__/jaxlib/mlir/dialects/ml_program.py",
"__main__/jaxlib/mlir/dialects/scf.py",
"__main__/jaxlib/mlir/dialects/sparse_tensor.py",
"__main__/jaxlib/mlir/dialects/stablehlo.py",