fix

Author: wsmoses

src/enzyme_ad/jax/primitives.py

@@ -425,9 +425,11 @@ def pass_pipeline(options):
 
 def resource_dir():
     import os
-
     dn = os.path.dirname(enzyme_call.__file__)
-    res = os.path.join(dn, "..", "..", "clang", "staging")
+    if os.getenv("ENZYME_BAZEL_NOWHEEL", None) is None:
+        res = os.path.join(dn, "..", "..", "..", "external", "llvm-project", "clang", "staging")
+    else:
+        res = os.path.join(dn, "..", "..", "clang", "staging")
     return res
 
 

test/lit.cfg.py

@@ -40,5 +40,6 @@
 ]
 path = os.path.pathsep.join(base_paths)  # + config.extra_paths)
 config.environment["PATH"] = path
+config.environment["ENZYME_TEST_NOWHEEL"] = "1"
 config.environment["PYTHONPATH"] = os.environ["PYTHONPATH"]
 config.substitutions.append(("python", sys.executable))