Skip to content

Commit df04e2f

Browse files
authored
feat: refactor pass pipeline options in python (#1036)
* feat: refactor pass pipeline options in python * fix: make inline an option
1 parent 756cf4c commit df04e2f

File tree

5 files changed

+452
-242
lines changed

5 files changed

+452
-242
lines changed

BUILD

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -167,7 +167,7 @@ py_wheel(
167167
python_requires = "=="+HERMETIC_PYTHON_VERSION+".*",
168168
strip_path_prefixes = ["src/"],
169169
summary = "Enzyme automatic differentiation tool.",
170-
version = "0.0.8",
170+
version = "0.0.10",
171171
deps = [
172172
":enzyme_jax_data",
173173
"//src/enzyme_ad/jax:enzyme_jax_internal",

src/enzyme_ad/jax/__init__.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -7,5 +7,6 @@
77
JaXPipeline,
88
optimize_module,
99
export,
10-
hlo_opts,
10+
full_optimization_pass_pipeline,
11+
optimization_passes,
1112
)

0 commit comments

Comments
 (0)