Merge pull request #3 from bytesnake/fix/llvm-headers

Add LLVM header requirements to README

Author: ZuseZ4

.packaging/rust/README.md

@@ -1,17 +1,26 @@
-# Goal
+# Enzyme build helper
+
+## Goal
+
 This repository will build enzyme/llvm/clang/rustc in the right configuration such that you can use it in combination with [oxide-enzyme](https://github.com/rust-ml/oxide-enzyme).
 
-# Requirements
-git  
-ninja  
-cmake  
-Rust (rustup)    
-~10GB free storage in $HOME/.config  
+## Requirements
+
+ - git  
+ - ninja  
+ - cmake  
+ - Rust (rustup)    
+ - ~10GB free storage in $HOME/.cache
+
+## Usage
+
+Build LLVM, the Rust toolchain and Enzyme with
 
-# Usage
-    $ git clone https://github.com/ZuseZ4/enzyme_build  
-    $ cd enzyme_build  
-    $ cargo run --release  
+```bash
+$ git clone https://github.com/ZuseZ4/enzyme_build  
+$ cd enzyme_build  
+$ cargo run --release  
+```
 
 Depending on your CPU this might take a few hours.  
 The build process will run enzyme tests, so your last output should look similar to these lines:
@@ -20,8 +29,8 @@ Testing Time: 0.63s
   Passed           : 240  
   Expectedly Failed:   5  
 
-# Extras
-- Q: Can I use some other location to store everyting?
+## Extras
+- Q: Can I use some other location to store everything?
 - A: We will add an option for that later. If it's urgent please open an issue or ping me (Manuel Drehwald) on the rust-ml Zulip.  
  
 - Q: It fails some (all) tests or the build breaks even earlier. Help?

.packaging/rust/src/generate_api.rs

@@ -12,6 +12,9 @@ pub fn generate_bindings() -> Result<(), String> {
 
     let bindings = bindgen::Builder::default()
         .header_contents("CApi.hpp", &content) // read it as .hpp so bindgen can ignore the class successfully
+        .clang_args(&[
+            format!("-I{}", utils::get_llvm_header_path().display())
+        ])
         //.blacklist_item("CustomFunctionForward")
         //.blacklist_item("DiffeGradientUtils")
         .allowlist_type("CConcreteType")

.packaging/rust/src/utils.rs

@@ -77,7 +77,9 @@ pub fn get_rustc_binary_path() -> PathBuf {
 pub fn get_llvm_build_path() -> PathBuf {
     get_rustc_platform_path().join("llvm").join("build")
 }
-
+pub fn get_llvm_header_path() -> PathBuf {
+    get_rustc_platform_path().join("llvm").join("include")
+}
 pub fn get_remote_enzyme_tarball_path() -> String {
     format!(
         "https://github.com/wsmoses/Enzyme/archive/refs/tags/v{}.tar.gz",