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

Move MHLO to XLA

Browse Source 
As part of the OpenXLA project, we're splitting XLA outside of TensorFlow.
MHLO belongs to OpenXLA and we're relocating it nested under XLA to allow the
split. Some further directory layout change will likely happen over time.

PiperOrigin-RevId: 464126676
This commit is contained in:
Mehdi Amini 2022-07-29 11:54:08 -07:00 committed by jax authors
parent 2178f339bd
commit 5a6cb438e8
2 changed files with 10 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
jaxlib/mlir/BUILD.bazel
View File

@ -44,7 +44,7 @@ symlink_inputs(
py_library(
name = "mlir",
deps = [
"//jaxlib/mlir/_mlir_libs:_mlir_libs",
"//jaxlib/mlir/_mlir_libs",
],
)
@ -106,7 +106,7 @@ symlink_inputs(
name = "mhlo_dialect",
rule = py_library,
symlinked_inputs = {"srcs": {"dialects": [
"@org_tensorflow//tensorflow/compiler/mlir/hlo:MhloOpsPyFiles",
"@org_tensorflow//tensorflow/compiler/xla/mlir_hlo:MhloOpsPyFiles",
]}},
deps = [
":core",

17
jaxlib/mlir/_mlir_libs/BUILD.bazel
View File

@ -17,7 +17,6 @@ load(
"py_extension",
"windows_cc_shared_mlir_library",
)
load("//jaxlib:symlink_files.bzl", "symlink_inputs")
package(
@ -75,9 +74,9 @@ py_extension(
py_extension(
name = "_mlirRegisterEverything",
srcs = [
"@llvm-project//mlir:lib/Bindings/Python/RegisterEverything.cpp",
"@llvm-project//mlir:include/mlir/Bindings/Python/PybindAdaptors.h",
"@llvm-project//mlir:include/mlir-c/Bindings/Python/Interop.h",
"@llvm-project//mlir:include/mlir/Bindings/Python/PybindAdaptors.h",
"@llvm-project//mlir:lib/Bindings/Python/RegisterEverything.cpp",
],
copts = COPTS,
deps = [
@ -110,7 +109,7 @@ symlink_inputs(
py_extension(
name = "_mlirHlo",
srcs = [
"@org_tensorflow//tensorflow/compiler/mlir/hlo:python/MlirHloModule.cc",
"@org_tensorflow//tensorflow/compiler/xla/mlir_hlo:python/MlirHloModule.cc",
],
copts = COPTS,
deps = [
@ -118,7 +117,7 @@ py_extension(
"@llvm-project//mlir:CAPIIRHeaders",
"@llvm-project//mlir:MLIRBindingsPythonHeaders",
"@local_config_python//:headers",
"@org_tensorflow//tensorflow/compiler/mlir/hlo:CAPIHeaders",
"@org_tensorflow//tensorflow/compiler/xla/mlir_hlo:CAPIHeaders",
"@pybind11",
],
)
@ -139,12 +138,12 @@ cc_library(
cc_library(
name = "jaxlib_mlir_capi_objects",
deps = [
"@llvm-project//mlir:MLIRBindingsPythonCAPIObjects",
"@llvm-project//mlir:CAPIConversionObjects",
"@llvm-project//mlir:CAPIRegisterEverythingObjects",
"@llvm-project//mlir:CAPISparseTensorObjects",
"@llvm-project//mlir:CAPITransformsObjects",
"@llvm-project//mlir:CAPIRegisterEverythingObjects",
"@llvm-project//mlir:CAPIConversionObjects",
"@org_tensorflow//tensorflow/compiler/mlir/hlo:CAPIObjects",
"@llvm-project//mlir:MLIRBindingsPythonCAPIObjects",
"@org_tensorflow//tensorflow/compiler/xla/mlir_hlo:CAPIObjects",
],
)