update submod; autodiff with callables

LuisaGroup · Oct 19, 2023 · dd04749 · dd04749
1 parent 52cc707
commit dd04749
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 6 deletions.
diff --git a/luisa_compute/tests/autodiff.rs b/luisa_compute/tests/autodiff.rs
@@ -1371,11 +1371,15 @@ fn autodiff_callable2() {
     let callable = Callable::<fn(Expr<f32>, Expr<f32>, Expr<i32>) -> Expr<f32>>::new(
         &device,
         track!(|x, y, t| {
-            switch::<Expr<f32>>(t)
-                .case(0, || x * 4.0)
-                .case(1, || x * 2.0)
-                .case(2, || y * 0.5)
-                .finish()
+            let ret = 0.0f32.var();
+            outline(|| {
+                *ret = switch::<Expr<f32>>(t)
+                    .case(0, || x * 4.0)
+                    .case(1, || x * 2.0)
+                    .case(2, || y * 0.5)
+                    .finish();
+            });
+            **ret
         }),
     );
     let kernel = Kernel::<fn()>::new(

diff --git a/luisa_compute_sys/LuisaCompute b/luisa_compute_sys/LuisaCompute
+1 −1		src/rust/luisa_compute_ir/src/transform/autodiff.rs
+21 −2		src/rust/luisa_compute_ir/src/transform/inliner.rs