Skip to content

Commit

Permalink
Update rules_python. Simplify and fix packaging of wheel.
Browse files Browse the repository at this point in the history
  • Loading branch information
ingomueller-net committed Nov 6, 2023
1 parent c4aace1 commit 10f8438
Show file tree
Hide file tree
Showing 3 changed files with 11 additions and 57 deletions.
7 changes: 1 addition & 6 deletions BUILD
Original file line number Diff line number Diff line change
Expand Up @@ -27,19 +27,14 @@ cc_library(
],
)

load("@rules_python//python:packaging.bzl", "py_wheel")

load(":package.bzl", "py_package")
load("@rules_python//python:packaging.bzl", "py_package", "py_wheel")

py_package(
name = "enzyme_jax_data",
# Only include these Python packages.
packages = ["//enzyme_jax:enzyme_call.so", "@llvm-project//clang:builtin_headers_gen"],
deps = [
"//enzyme_jax:enzyme_call",
"@llvm-project//clang:builtin_headers_gen",
],
prefix = "enzyme_jax/",
)

py_wheel(
Expand Down
12 changes: 10 additions & 2 deletions WORKSPACE
Original file line number Diff line number Diff line change
Expand Up @@ -42,8 +42,8 @@ http_archive(
patches = ["//:patches/xla.patch"],
)

PYRULES_COMMIT = "693a1587baf055979493565933f8f40225c00c6d"
PYRULES_SHA256 = "c493a9506b5e1ea99e3c22fb15e672cdd2a6fa19fd7c627ec5d485aced23e50a"
PYRULES_COMMIT = "fe33a4582c37499f3caeb49a07a78fc7948a8949"
PYRULES_SHA256 = "cfa6957832ae0e0c7ee2ccf455a888a291e8419ed8faf45f4420dd7414d5dd96"

http_archive(
name = "rules_python",
Expand All @@ -52,6 +52,14 @@ http_archive(
urls = ["https://github.com/bazelbuild/rules_python/archive/{commit}.tar.gz".format(commit = PYRULES_COMMIT)]
)

load("@rules_python//python:repositories.bzl", "py_repositories")

py_repositories()

load("@rules_python//python/pip_install:repositories.bzl", "pip_install_dependencies")

pip_install_dependencies()

ENZYME_COMMIT = "73ea8461c802689b2dfc2a6d9efea8ee9e506104"
ENZYME_SHA256 = "b1e5aa029b24985a018bed7012953eac7743391aace2b1f1e55a4a2cad9e343e"

Expand Down
49 changes: 0 additions & 49 deletions package.bzl

This file was deleted.

0 comments on commit 10f8438

Please sign in to comment.