From 2048a9eb39b7fb11cf4ce7eb3c5d0ac8f2b980e8 Mon Sep 17 00:00:00 2001 From: ReversedGravity Date: Fri, 22 Sep 2023 20:11:10 +0100 Subject: [PATCH] Added Expr - Var ops. --- luisa_compute/src/lang/ops/spread.rs | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/luisa_compute/src/lang/ops/spread.rs b/luisa_compute/src/lang/ops/spread.rs index c85500f..6657833 100644 --- a/luisa_compute/src/lang/ops/spread.rs +++ b/luisa_compute/src/lang/ops/spread.rs @@ -82,10 +82,14 @@ macro_rules! call_vector_fn_spread { $f!(['a, $($bounds)*] &'a $T: |x| $Vsplat(*x), Expr<$Vt>: |x| x => Expr<$Vt>); $f!([$($bounds)*] Expr<$T>: |x| $Vsplat(x), Expr<$Vt>: |x| x => Expr<$Vt>); $f!(['a, $($bounds)*] &'a Expr<$T>: |x| $Vsplat(x), Expr<$Vt>: |x| x => Expr<$Vt>); + $f!([$($bounds)*] Var<$T>: |x| $Vsplat(x), Expr<$Vt>: |x| x => Expr<$Vt>); + $f!(['a, $($bounds)*] &'a Var<$T>: |x| $Vsplat(x), Expr<$Vt>: |x| x => Expr<$Vt>); $f!(['b, $($bounds)*] $T: |x| $Vsplat(x), &'b Expr<$Vt>: |x| x.clone() => Expr<$Vt>); $f!(['a, 'b, $($bounds)*] &'a $T: |x| $Vsplat(*x), &'b Expr<$Vt>: |x| x.clone() => Expr<$Vt>); $f!(['b, $($bounds)*] Expr<$T>: |x| $Vsplat(x), &'b Expr<$Vt>: |x| x.clone() => Expr<$Vt>); $f!(['a, 'b, $($bounds)*] &'a Expr<$T>: |x| $Vsplat(x), &'b Expr<$Vt>: |x| x.clone() => Expr<$Vt>); + $f!(['b, $($bounds)*] Var<$T>: |x| $Vsplat(x), &'b Expr<$Vt>: |x| x.clone() => Expr<$Vt>); + $f!(['a, 'b, $($bounds)*] &'a Var<$T>: |x| $Vsplat(x), &'b Expr<$Vt>: |x| x.clone() => Expr<$Vt>); $f!([$($bounds)*] Expr<$T>: |x| $Vsplat(x), $Vt: |x| x.expr() => Expr<$Vt>); $f!(['a, $($bounds)*] &'a Expr<$T>: |x| $Vsplat(x), $Vt: |x| x.expr() => Expr<$Vt>);