added signum for floats

luisa_compute/src/lang/ops/impls.rs

@@ -236,6 +236,16 @@ where
     fn sin_cos(&self) -> (Self, Self) {
         (self.sin(), self.cos())
     }
+    fn signum(&self) -> Self {
+        let self_node = self.node().get();
+        <Self as FromNode>::from_node(
+            __current_scope(|b| {
+                let one = b.const_(Const::One(<X as TypeOf>::type_()));
+                b.call(Func::Copysign, &[one, self_node], <X as TypeOf>::type_())
+            })
+            .into(),
+        )
+    }
 }
 impl<const N: usize, X: Floating> NormExpr for Expr<Vector<X, N>>
 where

luisa_compute/src/lang/ops/spread.rs

@@ -286,6 +286,7 @@ where
         Expr::<T>::_mul_add(self, S::lift_self(mul), S::lift_other(add))
     }
 }
+
 impl<T, S> FloatCopySignExpr<S> for T
 where
     T: SpreadOps<S>,

luisa_compute/src/lang/ops/traits.rs

@@ -244,6 +244,7 @@ pub trait FloatExpr: Sized {
     fn cube(&self) -> Self;
     fn recip(&self) -> Self;
     fn sin_cos(&self) -> (Self, Self);
+    fn signum(&self) -> Self;
 }
 
 pub trait ReduceExpr: Sized {

luisa_compute/src/lib.rs

@@ -198,7 +198,11 @@ impl Context {
     pub fn create_device<D: IntoDeviceName>(&self, device: D) -> Device {
         self.create_device_with_config(device, serde_json::json!({}))
     }
-    pub fn create_device_with_config<D: IntoDeviceName>(&self, device: D, config: serde_json::Value) -> Device {
+    pub fn create_device_with_config<D: IntoDeviceName>(
+        &self,
+        device: D,
+        config: serde_json::Value,
+    ) -> Device {
         let backend = self.inner.create_device(&device.into_device_name(), config);
         let default_stream = backend.create_stream(api::StreamTag::Graphics);
         Device {