commit 44ac31d554154efa7ba0c4753e5a3ebfaf1a8618
author Geoffrey Martin-Noble <gcmn@google.com> Fri Apr 10 11:16:16 2020 -0700
committer Copybara-Service <copybara-worker@google.com> Fri Apr 10 11:17:22 2020 -0700
tree e47c01f0df7daaf57649c0459f15ad77907e1efa
parent 5fd6dcf44f5a0c8330e1a2ac9f71f1f8127ef881

Re-enable LLVM JIT in check framework

PiperOrigin-RevId: 305909346

build_tools/bazel/iree_check_test.bzl

diff --git a/build_tools/bazel/iree_check_test.bzl b/build_tools/bazel/iree_check_test.bzl
index 2469b25..ff1fc21 100644
--- a/build_tools/bazel/iree_check_test.bzl
+++ b/build_tools/bazel/iree_check_test.bzl
@@ -20,6 +20,7 @@
 ALL_TARGET_BACKENDS_AND_DRIVERS = [
     ("vmla", "vmla"),
     ("vulkan-spirv", "vulkan"),
+    ("llvm-ir", "llvm"),
 ]
 
 def iree_check_test(name, src, target_backend, driver, args = [], tags = [], **kwargs):

build_tools/cmake/iree_check_test.cmake

diff --git a/build_tools/cmake/iree_check_test.cmake b/build_tools/cmake/iree_check_test.cmake
index d6535f8..54c42b4 100644
--- a/build_tools/cmake/iree_check_test.cmake
+++ b/build_tools/cmake/iree_check_test.cmake
@@ -139,8 +139,8 @@
   endif()
 
   if(NOT DEFINED _RULE_TARGET_BACKENDS AND NOT DEFINED _RULE_DRIVERS)
-    set(_RULE_TARGET_BACKENDS "vmla" "vulkan-spirv")
-    set(_RULE_DRIVERS "vmla" "vulkan")
+    set(_RULE_TARGET_BACKENDS "vmla" "vulkan-spirv" "llvm-ir")
+    set(_RULE_DRIVERS "vmla" "vulkan" "llvm")
   endif()
 
   list(LENGTH _RULE_TARGET_BACKENDS _TARGET_BACKEND_COUNT)

iree/test/e2e/xla/BUILD

diff --git a/iree/test/e2e/xla/BUILD b/iree/test/e2e/xla/BUILD
index 06e7401..ca87829 100644
--- a/iree/test/e2e/xla/BUILD
+++ b/iree/test/e2e/xla/BUILD
@@ -41,4 +41,8 @@
 iree_check_test_suite(
     name = "check",
     srcs = CHECK_TESTS,
+    target_backends_and_drivers = [
+        ("vmla", "vmla"),
+        ("vulkan-spirv", "vulkan"),
+    ],
 )

iree/test/e2e/xla/CMakeLists.txt

diff --git a/iree/test/e2e/xla/CMakeLists.txt b/iree/test/e2e/xla/CMakeLists.txt
index abc700f..f8cdc2c 100644
--- a/iree/test/e2e/xla/CMakeLists.txt
+++ b/iree/test/e2e/xla/CMakeLists.txt
@@ -32,4 +32,10 @@
     check
   SRCS
     "reverse.mlir"
+  TARGET_BACKENDS
+    "vmla"
+    "vulkan-spirv"
+  DRIVERS
+    "vmla"
+    "vulkan"
 )