From 0d950e715e0ee606c9af8c862fbd34918a07ed68 Mon Sep 17 00:00:00 2001
From: William Moses <gh@wsmoses.com>
Date: Mon, 9 Sep 2024 20:01:38 -0400
Subject: [PATCH] Update BUILD

---
 enzyme/test/Integration/BUILD | 38 +++++++++++++++++------------------
 1 file changed, 19 insertions(+), 19 deletions(-)

diff --git a/enzyme/test/Integration/BUILD b/enzyme/test/Integration/BUILD
index 2c6aa8116a6..e57bad19ce1 100644
--- a/enzyme/test/Integration/BUILD
+++ b/enzyme/test/Integration/BUILD
@@ -2,6 +2,24 @@
 
 load("@llvm-project//llvm:lit_test.bzl", "lit_test")
 
+exclude_tests = [
+    "**/*omp*.c",
+]
+
+stdlib_tests = [
+    "**/gh_issue_1904.cpp",
+    "**/CppSugar/sret.cpp",
+    "**/ReverseMode/fbuff.cpp",
+    "**/*blas*.cpp",
+    "**/*ForwardMode/rosenbrock.cpp",
+    "**/*ReverseMode/map.cpp",
+    "**/*ReverseMode/multivecmax.cpp",
+    "**/*ReverseMode/vecmax.cpp",
+    "**/*Sparse/*",
+    "**/*eigen*.cpp",
+    "**/*integrate*.cpp",
+]
+
 [
     lit_test(
         name = "%s.test" % src,
@@ -18,30 +36,12 @@ load("@llvm-project//llvm:lit_test.bzl", "lit_test")
             "@llvm-project//llvm:lli",
             "@llvm-project//llvm:not",
         ] + glob(["**/*.h"]),
-        driver = "//test:run_lit",
-        toolchains = [
-            "//tools/cpp:cc_flags",
-            "//tools/cpp:current_cc_toolchain",
-        ],
     )
     for src in glob(
         [
             "**/*.c",
             "**/*.cpp",
         ],
-        exclude = [
-            "**/*omp*.c",
-            "**/gh_issue_1904.cpp",
-            "**/CppSugar/sret.cpp",
-            "**/ReverseMode/fbuff.cpp",
-            "**/*blas*.cpp",
-            "**/*ForwardMode/rosenbrock.cpp",
-            "**/*ReverseMode/map.cpp",
-            "**/*ReverseMode/multivecmax.cpp",
-            "**/*ReverseMode/vecmax.cpp",
-            "**/*Sparse/*",
-            "**/*eigen*.cpp",
-            "**/*integrate*.cpp",
-        ],
+        exclude = exclude_tests,
     )
 ]