diff --git a/src/enzyme_ad/jax/primitives.py b/src/enzyme_ad/jax/primitives.py index 2fffc7675..99a9cfa8e 100644 --- a/src/enzyme_ad/jax/primitives.py +++ b/src/enzyme_ad/jax/primitives.py @@ -25,7 +25,8 @@ def resource_dir(): import os dn = os.path.dirname(enzyme_call.__file__) - return os.path.join(dn, "..", "clang", "staging") + res = os.path.join(dn, "..", "..", "clang", "staging") + return res def cflags(): import platform