move sources under enzyme_jax (#15)

This makes it easier to import this project into other bazel-built projects.
EnzymeAD · Dec 1, 2023 · 3bf54c4 · 3bf54c4
1 parent 8107881
commit 3bf54c4
Show file tree

Hide file tree

Showing 4 changed files with 30 additions and 26 deletions.
diff --git a/BUILD b/BUILD
@@ -1,3 +1,5 @@
+load("@rules_python//python:packaging.bzl", "py_wheel")
+load(":package.bzl", "py_package")
 
 licenses(["notice"])
 
@@ -6,30 +8,6 @@ package(
     default_visibility = ["//:__subpackages__"],
 )
 
-cc_library(
-    name = "clang_compile",
-    srcs = ["clang_compile.cc"],
-    hdrs = ["clang_compile.h"],
-    deps = [
-        "@pybind11",
-        "@llvm-project//clang:ast",
-        "@llvm-project//clang:basic",
-        "@llvm-project//clang:driver",
-        "@llvm-project//clang:frontend",
-        "@llvm-project//clang:frontend_tool",
-        "@llvm-project//clang:lex",
-        "@llvm-project//clang:serialization",
-        "@llvm-project//llvm:Support",
-        "@llvm-project//llvm:Core",
-        "@llvm-project//llvm:IRReader",
-        "@llvm-project//llvm:OrcJIT",
-        "@enzyme//:EnzymeStatic"
-    ],
-)
-
-load("@rules_python//python:packaging.bzl", "py_wheel")
-load(":package.bzl", "py_package")
-
 py_package(
     name = "enzyme_jax_data",
     deps = [

diff --git a/enzyme_jax/BUILD b/enzyme_jax/BUILD
@@ -1,6 +1,32 @@
-
 load("@pybind11_bazel//:build_defs.bzl", "pybind_extension", "pybind_library")
 
+licenses(["notice"])
+
+package(
+    default_visibility = ["//:__subpackages__"],
+)
+
+cc_library(
+    name = "clang_compile",
+    srcs = ["clang_compile.cc"],
+    hdrs = ["clang_compile.h"],
+    deps = [
+        "@pybind11",
+        "@llvm-project//clang:ast",
+        "@llvm-project//clang:basic",
+        "@llvm-project//clang:driver",
+        "@llvm-project//clang:frontend",
+        "@llvm-project//clang:frontend_tool",
+        "@llvm-project//clang:lex",
+        "@llvm-project//clang:serialization",
+        "@llvm-project//llvm:Support",
+        "@llvm-project//llvm:Core",
+        "@llvm-project//llvm:IRReader",
+        "@llvm-project//llvm:OrcJIT",
+        "@enzyme//:EnzymeStatic"
+    ],
+)
+
 py_library(
     name = "enzyme_jax_internal",
     srcs = ["primitives.py", "__init__.py"],
@@ -72,7 +98,7 @@ pybind_extension(
     deps = [
         "@llvm-project//llvm:Support",
         "@llvm-project//llvm:OrcJIT",
-        "//:clang_compile",
+        ":clang_compile",
         ":compile_with_xla",
         "@com_google_absl//absl/status"
     ],

diff --git a/clang_compile.cc → enzyme_jax/clang_compile.cc b/clang_compile.cc → enzyme_jax/clang_compile.cc
diff --git a/clang_compile.h → enzyme_jax/clang_compile.h b/clang_compile.h → enzyme_jax/clang_compile.h