Merge branch 'openxla:main' into main

.bazelrc

@@ -21,3 +21,14 @@ build --features=layering_check
 # Set the default compiler to the `clang` binary on the `PATH`.
 # TODO(fzakaria): Make this a toolchain or hermetic somehow
 build --repo_env=CC=clang
+
+# Disabling runfiles links drastically increases performance in slow disk IO situations
+# Do not build runfile trees by default. If an execution strategy relies on runfile
+# symlink tree, the tree is created on-demand. See: https://github.com/bazelbuild/bazel/issues/6627
+# and https://github.com/bazelbuild/bazel/commit/03246077f948f2790a83520e7dccc2625650e6df
+build --nobuild_runfile_links
+test --nobuild_runfile_links
+
+# https://bazel.build/reference/command-line-reference#flag--legacy_external_runfiles
+build --nolegacy_external_runfiles
+test --nolegacy_external_runfiles

.github/workflows/buildAndTestBazel.yml

@@ -59,7 +59,9 @@ jobs:
       uses: actions/cache@v3
       with:
         path: "~/.cache/bazel"
-        key: ${{ runner.os }}-stablehlo_build_assets-${{ steps.llvm-version.outputs.version }}-bazelbuild
+        key: ${{ runner.os }}-stablehlo_bazelbuild-${{ hashFiles('.bazelversion', '.bazelrc', 'WORKSPACE', 'WORKSPACE.bazel', 'MODULE.bazel') }}
+        restore-keys: |
+          ${{ runner.os }}-stablehlo_bazelbuild-
 
     - name: Build StableHLO
       shell: bash

BUILD.bazel

@@ -431,6 +431,7 @@ cc_library(
         ":reference_ops",
         ":reference_process",
         ":reference_scope",
+        ":reference_tensor",
         ":reference_value",
         ":register",
         "@llvm-project//llvm:Support",
@@ -927,10 +928,13 @@ cc_library(
     srcs = [
         "stablehlo/transforms/ChloLegalizeToStablehlo.cpp",
         "stablehlo/transforms/PassPipelines.cpp",
+        "stablehlo/transforms/ShapeLegalizeToStablehlo.cpp",
         "stablehlo/transforms/StablehloAggressiveSimplification.cpp",
         "stablehlo/transforms/StablehloCanonicalizeDynamism.cpp",
         "stablehlo/transforms/StablehloInstrumentWithProbe.cpp",
+        "stablehlo/transforms/StablehloLegalizeCompositeToCall.cpp",
         "stablehlo/transforms/StablehloLegalizeToVhlo.cpp",
+        "stablehlo/transforms/StablehloRefineArguments.cpp",
         "stablehlo/transforms/StablehloRefineShapes.cpp",
         "stablehlo/transforms/VhloLegalizeToStablehlo.cpp",
         "stablehlo/transforms/VhloToVersion.cpp",
@@ -955,6 +959,8 @@ cc_library(
         ":vhlo_ops",
         ":vhlo_types",
         "@llvm-project//llvm:Support",
+        "@llvm-project//mlir:ArithDialect",
+        "@llvm-project//mlir:AsmParser",
         "@llvm-project//mlir:CommonFolders",
         "@llvm-project//mlir:ComplexDialect",
         "@llvm-project//mlir:FuncDialect",
@@ -1031,6 +1037,7 @@ cc_library(
         ":base",
         ":stablehlo_assembly_format",
         "@llvm-project//llvm:Support",
+        "@llvm-project//mlir:FuncDialect",
         "@llvm-project//mlir:IR",
         "@llvm-project//mlir:InferTypeOpInterface",
         "@llvm-project//mlir:QuantOps",
@@ -1127,6 +1134,7 @@ cc_binary(
         ":register",
         ":stablehlo_passes",
         ":tosa_passes",
+        "//stablehlo/tests:check_ops",
         "//stablehlo/tests:test_utils",
         "@llvm-project//mlir:AllExtensions",
         "@llvm-project//mlir:AllPassesAndDialects",

CMakeLists.txt

@@ -187,3 +187,4 @@ add_custom_target(check-stablehlo-quick)
 add_dependencies(check-stablehlo-ci check-stablehlo-quick)
 
 add_subdirectory(stablehlo)
+add_subdirectory(examples)

CMakePresets.json

@@ -0,0 +1,24 @@
+{
+    "version": 6,
+    "configurePresets": [
+        {
+          "name": "debug",
+          "displayName": "Debug w/ ccache",
+          "generator": "Ninja",
+          "binaryDir": "build/",
+          "cacheVariables": {
+            "CMAKE_BUILD_TYPE": "Debug",
+            "LLVM_ENABLE_ASSERTIONS": "ON",
+            "LLVM_ENABLE_LLD": "ON",
+            "STABLEHLO_ENABLE_BINDINGS_PYTHON" : "OFF",
+            "STABLEHLO_ENABLE_SPLIT_DWARF": "ON",
+            "CMAKE_CXX_COMPILER_LAUNCHER": "ccache",
+            "CMAKE_CXX_COMPILER": "clang++",
+            "CMAKE_C_COMPILER_LAUNCHER": "ccache",
+            "CMAKE_C_COMPILER": "clang",
+            "CMAKE_EXPORT_COMPILE_COMMANDS": "ON",
+            "MLIR_DIR": "${sourceDir}/llvm-build/lib/cmake/mlir"
+          }
+        }
+      ]
+  }

CODE_OF_CONDUCT.md

@@ -67,7 +67,7 @@ threatening or harassing, report it. We are dedicated to providing an environmen
 where participants feel welcome and safe.
 
 Reports should be directed to StableHLO's Project Stewards,
-Thea Lamkin (thealamkin@google.com) and Eugene Burmako (burmako@google.com).
+Kevin Gleason (gleasonk@google.com) and Farid Zakaria (fmzakari@google.com).
 It is the Project Steward’s duty to receive and address reported violations of
 the code of conduct.