Compile Floretta to Wasm

samestep · Nov 20, 2024 · b1e5a45 · b1e5a45
1 parent 6b5d7c8
commit b1e5a45
Show file tree

Hide file tree

Showing 5 changed files with 31 additions and 0 deletions.
diff --git a/.github/workflows/build.yml b/.github/workflows/build.yml
@@ -11,3 +11,8 @@ jobs:
     steps:
       - uses: actions/checkout@v4
       - run: cargo test
+      - run: cargo build --package=floretta-wasm --target=wasm32-unknown-unknown --release
+      - uses: actions/upload-artifact@v4
+        with:
+          name: floretta.wasm
+          path: target/wasm32-unknown-unknown/release/floretta_wasm.wasm
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/crates/floretta-wasm/Cargo.toml b/crates/floretta-wasm/Cargo.toml
@@ -0,0 +1,11 @@
+[package]
+name = "floretta-wasm"
+private = true
+version.workspace = true
+edition.workspace = true
+
+[lib]
+crate-type = ["cdylib"]
+
+[dependencies]
+floretta.workspace = true
diff --git a/crates/floretta-wasm/src/lib.rs b/crates/floretta-wasm/src/lib.rs
@@ -0,0 +1,4 @@
+#[no_mangle]
+fn autodiff(wasm_module: &[u8]) -> Result<Vec<u8>, floretta::Error> {
+    floretta::Autodiff::new().transform(wasm_module)
+}
diff --git a/rust-toolchain.toml b/rust-toolchain.toml
@@ -0,0 +1,4 @@
+[toolchain]
+channel = "stable"
+profile = "default"
+targets = ["wasm32-unknown-unknown"]