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

Integrate StableHLO at openxla/stablehlo@14691ce

Browse Source 
Manual changes:
  * stablehlo/integrations/python/mlir/dialects/stablehlo.py: to keep around get_earliest_forward_compatible_version while it's still used in JAX.

PiperOrigin-RevId: 533140501
This commit is contained in:
Eugene Burmako 2023-05-18 08:41:42 -07:00 committed by jax authors
parent b81b79c1b0
commit 8696bef218
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -159,6 +159,8 @@ py_extension(
py_extension(
name = "_stablehlo",
srcs = [
"@stablehlo//:stablehlo/integrations/python/PortableApi.cpp",
"@stablehlo//:stablehlo/integrations/python/PortableApi.h",
"@stablehlo//:stablehlo/integrations/python/StablehloModule.cpp",
],
copts = COPTS,
@ -170,8 +172,8 @@ py_extension(
"@local_config_python//:headers",
"@pybind11",
"@stablehlo//:stablehlo_capi_headers",
"@stablehlo//:stablehlo_portable_api",
"@stablehlo//:stablehlo_serialization",
"@stablehlo//:version",
],
)