update dynstack

sarah-quinones · Sep 29, 2024 · 7e5435d · 7e5435d
1 parent 0d5bba0
commit 7e5435d
Show file tree

Hide file tree

Showing 17 changed files with 97 additions and 92 deletions.
diff --git a/src/linalg/cholesky/piv_llt/mod.rs b/src/linalg/cholesky/piv_llt/mod.rs
@@ -48,7 +48,7 @@ pub mod compute {
         perm: &'out mut [I],
         perm_inv: &'out mut [I],
         parallelism: Parallelism,
-        stack: PodStack<'_>,
+        stack: &mut PodStack,
         params: PivLltParams,
     ) -> Result<(PivLltInfo, PermRef<'out, I>), CholeskyError> {
         assert!(a.nrows() == a.ncols());

diff --git a/src/linop/bicgstab.rs b/src/linop/bicgstab.rs
@@ -138,7 +138,7 @@ pub fn bicgstab<E: ComplexField>(
     params: BicgParams<E>,
     callback: impl FnMut(MatRef<'_, E>),
     parallelism: Parallelism,
-    stack: PodStack<'_>,
+    stack: &mut PodStack,
 ) -> Result<BicgInfo<E>, BicgError<E>> {
     #[track_caller]
     fn implementation<E: ComplexField>(
@@ -150,7 +150,7 @@ pub fn bicgstab<E: ComplexField>(
         params: BicgParams<E>,
         callback: &mut dyn FnMut(MatRef<'_, E>),
         parallelism: Parallelism,
-        stack: PodStack<'_>,
+        stack: &mut PodStack,
     ) -> Result<BicgInfo<E>, BicgError<E>> {
         let mut x = out;
         let A = mat;

diff --git a/src/linop/conjugate_gradient.rs b/src/linop/conjugate_gradient.rs
@@ -133,7 +133,7 @@ pub fn conjugate_gradient<E: ComplexField>(
     params: CgParams<E>,
     callback: impl FnMut(MatRef<'_, E>),
     parallelism: Parallelism,
-    stack: PodStack<'_>,
+    stack: &mut PodStack,
 ) -> Result<CgInfo<E>, CgError<E>> {
     #[track_caller]
     fn implementation<E: ComplexField>(
@@ -145,7 +145,7 @@ pub fn conjugate_gradient<E: ComplexField>(
         params: CgParams<E>,
         callback: &mut dyn FnMut(MatRef<'_, E>),
         parallelism: Parallelism,
-        mut stack: PodStack<'_>,
+        mut stack: &mut PodStack,
     ) -> Result<CgInfo<E>, CgError<E>> {
         assert!(A.nrows() == A.ncols());
 

diff --git a/src/linop/linop_impl/csc_matmut.rs b/src/linop/linop_impl/csc_matmut.rs
@@ -33,7 +33,7 @@ impl<E: ComplexField, I: Index, ViewE: Conjugate<Canonical = E>> LinOp<E>
         out: MatMut<'_, E>,
         rhs: MatRef<'_, E>,
         parallelism: Parallelism,
-        stack: PodStack<'_>,
+        stack: &mut PodStack,
     ) {
         self.as_ref().apply(out, rhs, parallelism, stack)
     }
@@ -45,7 +45,7 @@ impl<E: ComplexField, I: Index, ViewE: Conjugate<Canonical = E>> LinOp<E>
         out: MatMut<'_, E>,
         rhs: MatRef<'_, E>,
         parallelism: Parallelism,
-        stack: PodStack<'_>,
+        stack: &mut PodStack,
     ) {
         self.as_ref().conj_apply(out, rhs, parallelism, stack)
     }
@@ -70,7 +70,7 @@ impl<E: ComplexField, I: Index, ViewE: Conjugate<Canonical = E>> BiLinOp<E>
         out: MatMut<'_, E>,
         rhs: MatRef<'_, E>,
         parallelism: Parallelism,
-        stack: PodStack<'_>,
+        stack: &mut PodStack,
     ) {
         self.as_ref().transpose_apply(out, rhs, parallelism, stack)
     }
@@ -82,7 +82,7 @@ impl<E: ComplexField, I: Index, ViewE: Conjugate<Canonical = E>> BiLinOp<E>
         out: MatMut<'_, E>,
         rhs: MatRef<'_, E>,
         parallelism: Parallelism,
-        stack: PodStack<'_>,
+        stack: &mut PodStack,
     ) {
         self.as_ref().adjoint_apply(out, rhs, parallelism, stack)
     }

diff --git a/src/linop/linop_impl/csc_matown.rs b/src/linop/linop_impl/csc_matown.rs
@@ -33,7 +33,7 @@ impl<E: ComplexField, I: Index, ViewE: Conjugate<Canonical = E>> LinOp<E>
         out: MatMut<'_, E>,
         rhs: MatRef<'_, E>,
         parallelism: Parallelism,
-        stack: PodStack<'_>,
+        stack: &mut PodStack,
     ) {
         self.as_ref().apply(out, rhs, parallelism, stack)
     }
@@ -45,7 +45,7 @@ impl<E: ComplexField, I: Index, ViewE: Conjugate<Canonical = E>> LinOp<E>
         out: MatMut<'_, E>,
         rhs: MatRef<'_, E>,
         parallelism: Parallelism,
-        stack: PodStack<'_>,
+        stack: &mut PodStack,
     ) {
         self.as_ref().conj_apply(out, rhs, parallelism, stack)
     }
@@ -70,7 +70,7 @@ impl<E: ComplexField, I: Index, ViewE: Conjugate<Canonical = E>> BiLinOp<E>
         out: MatMut<'_, E>,
         rhs: MatRef<'_, E>,
         parallelism: Parallelism,
-        stack: PodStack<'_>,
+        stack: &mut PodStack,
     ) {
         self.as_ref().transpose_apply(out, rhs, parallelism, stack)
     }
@@ -82,7 +82,7 @@ impl<E: ComplexField, I: Index, ViewE: Conjugate<Canonical = E>> BiLinOp<E>
         out: MatMut<'_, E>,
         rhs: MatRef<'_, E>,
         parallelism: Parallelism,
-        stack: PodStack<'_>,
+        stack: &mut PodStack,
     ) {
         self.as_ref().adjoint_apply(out, rhs, parallelism, stack)
     }

diff --git a/src/linop/linop_impl/csc_matref.rs b/src/linop/linop_impl/csc_matref.rs
@@ -34,7 +34,7 @@ impl<E: ComplexField, I: Index, ViewE: Conjugate<Canonical = E>> LinOp<E>
         out: MatMut<'_, E>,
         rhs: MatRef<'_, E>,
         parallelism: Parallelism,
-        stack: PodStack<'_>,
+        stack: &mut PodStack,
     ) {
         _ = stack;
         crate::sparse::linalg::matmul::sparse_dense_matmul(
@@ -54,7 +54,7 @@ impl<E: ComplexField, I: Index, ViewE: Conjugate<Canonical = E>> LinOp<E>
         out: MatMut<'_, E>,
         rhs: MatRef<'_, E>,
         parallelism: Parallelism,
-        stack: PodStack<'_>,
+        stack: &mut PodStack,
     ) {
         _ = stack;
         let this = self.conjugate();
@@ -89,7 +89,7 @@ impl<E: ComplexField, I: Index, ViewE: Conjugate<Canonical = E>> BiLinOp<E>
         out: MatMut<'_, E>,
         rhs: MatRef<'_, E>,
         parallelism: Parallelism,
-        stack: PodStack<'_>,
+        stack: &mut PodStack,
     ) {
         _ = stack;
         let this = self.transpose();
@@ -110,7 +110,7 @@ impl<E: ComplexField, I: Index, ViewE: Conjugate<Canonical = E>> BiLinOp<E>
         out: MatMut<'_, E>,
         rhs: MatRef<'_, E>,
         parallelism: Parallelism,
-        stack: PodStack<'_>,
+        stack: &mut PodStack,
     ) {
         _ = stack;
         let this = self.adjoint();

diff --git a/src/linop/linop_impl/csr_matmut.rs b/src/linop/linop_impl/csr_matmut.rs
@@ -33,7 +33,7 @@ impl<E: ComplexField, I: Index, ViewE: Conjugate<Canonical = E>> LinOp<E>
         out: MatMut<'_, E>,
         rhs: MatRef<'_, E>,
         parallelism: Parallelism,
-        stack: PodStack<'_>,
+        stack: &mut PodStack,
     ) {
         self.as_ref().apply(out, rhs, parallelism, stack)
     }
@@ -45,7 +45,7 @@ impl<E: ComplexField, I: Index, ViewE: Conjugate<Canonical = E>> LinOp<E>
         out: MatMut<'_, E>,
         rhs: MatRef<'_, E>,
         parallelism: Parallelism,
-        stack: PodStack<'_>,
+        stack: &mut PodStack,
     ) {
         self.as_ref().conj_apply(out, rhs, parallelism, stack)
     }
@@ -70,7 +70,7 @@ impl<E: ComplexField, I: Index, ViewE: Conjugate<Canonical = E>> BiLinOp<E>
         out: MatMut<'_, E>,
         rhs: MatRef<'_, E>,
         parallelism: Parallelism,
-        stack: PodStack<'_>,
+        stack: &mut PodStack,
     ) {
         self.as_ref().transpose_apply(out, rhs, parallelism, stack)
     }
@@ -82,7 +82,7 @@ impl<E: ComplexField, I: Index, ViewE: Conjugate<Canonical = E>> BiLinOp<E>
         out: MatMut<'_, E>,
         rhs: MatRef<'_, E>,
         parallelism: Parallelism,
-        stack: PodStack<'_>,
+        stack: &mut PodStack,
     ) {
         self.as_ref().adjoint_apply(out, rhs, parallelism, stack)
     }

diff --git a/src/linop/linop_impl/csr_matown.rs b/src/linop/linop_impl/csr_matown.rs
@@ -33,7 +33,7 @@ impl<E: ComplexField, I: Index, ViewE: Conjugate<Canonical = E>> LinOp<E>
         out: MatMut<'_, E>,
         rhs: MatRef<'_, E>,
         parallelism: Parallelism,
-        stack: PodStack<'_>,
+        stack: &mut PodStack,
     ) {
         self.as_ref().apply(out, rhs, parallelism, stack)
     }
@@ -45,7 +45,7 @@ impl<E: ComplexField, I: Index, ViewE: Conjugate<Canonical = E>> LinOp<E>
         out: MatMut<'_, E>,
         rhs: MatRef<'_, E>,
         parallelism: Parallelism,
-        stack: PodStack<'_>,
+        stack: &mut PodStack,
     ) {
         self.as_ref().conj_apply(out, rhs, parallelism, stack)
     }
@@ -70,7 +70,7 @@ impl<E: ComplexField, I: Index, ViewE: Conjugate<Canonical = E>> BiLinOp<E>
         out: MatMut<'_, E>,
         rhs: MatRef<'_, E>,
         parallelism: Parallelism,
-        stack: PodStack<'_>,
+        stack: &mut PodStack,
     ) {
         self.as_ref().transpose_apply(out, rhs, parallelism, stack)
     }
@@ -82,7 +82,7 @@ impl<E: ComplexField, I: Index, ViewE: Conjugate<Canonical = E>> BiLinOp<E>
         out: MatMut<'_, E>,
         rhs: MatRef<'_, E>,
         parallelism: Parallelism,
-        stack: PodStack<'_>,
+        stack: &mut PodStack,
     ) {
         self.as_ref().adjoint_apply(out, rhs, parallelism, stack)
     }

diff --git a/src/linop/linop_impl/csr_matref.rs b/src/linop/linop_impl/csr_matref.rs
@@ -34,7 +34,7 @@ impl<E: ComplexField, I: Index, ViewE: Conjugate<Canonical = E>> LinOp<E>
         out: MatMut<'_, E>,
         rhs: MatRef<'_, E>,
         parallelism: Parallelism,
-        stack: PodStack<'_>,
+        stack: &mut PodStack,
     ) {
         _ = stack;
         crate::sparse::linalg::matmul::dense_sparse_matmul(
@@ -54,7 +54,7 @@ impl<E: ComplexField, I: Index, ViewE: Conjugate<Canonical = E>> LinOp<E>
         out: MatMut<'_, E>,
         rhs: MatRef<'_, E>,
         parallelism: Parallelism,
-        stack: PodStack<'_>,
+        stack: &mut PodStack,
     ) {
         _ = stack;
         let this = self.conjugate();
@@ -89,7 +89,7 @@ impl<E: ComplexField, I: Index, ViewE: Conjugate<Canonical = E>> BiLinOp<E>
         out: MatMut<'_, E>,
         rhs: MatRef<'_, E>,
         parallelism: Parallelism,
-        stack: PodStack<'_>,
+        stack: &mut PodStack,
     ) {
         _ = stack;
         let this = self.transpose();
@@ -110,7 +110,7 @@ impl<E: ComplexField, I: Index, ViewE: Conjugate<Canonical = E>> BiLinOp<E>
         out: MatMut<'_, E>,
         rhs: MatRef<'_, E>,
         parallelism: Parallelism,
-        stack: PodStack<'_>,
+        stack: &mut PodStack,
     ) {
         _ = stack;
         let this = self.adjoint();

diff --git a/src/linop/linop_impl/diagmut.rs b/src/linop/linop_impl/diagmut.rs
@@ -30,7 +30,7 @@ impl<E: ComplexField, ViewE: Conjugate<Canonical = E>> LinOp<E> for DiagMut<'_,
         out: MatMut<'_, E>,
         rhs: MatRef<'_, E>,
         parallelism: Parallelism,
-        stack: PodStack<'_>,
+        stack: &mut PodStack,
     ) {
         self.as_ref().apply(out, rhs, parallelism, stack)
     }
@@ -40,7 +40,7 @@ impl<E: ComplexField, ViewE: Conjugate<Canonical = E>> LinOp<E> for DiagMut<'_,
         out: MatMut<'_, E>,
         rhs: MatRef<'_, E>,
         parallelism: Parallelism,
-        stack: PodStack<'_>,
+        stack: &mut PodStack,
     ) {
         self.as_ref().conj_apply(out, rhs, parallelism, stack)
     }
@@ -61,7 +61,7 @@ impl<E: ComplexField, ViewE: Conjugate<Canonical = E>> BiLinOp<E> for DiagMut<'_
         out: MatMut<'_, E>,
         rhs: MatRef<'_, E>,
         parallelism: Parallelism,
-        stack: PodStack<'_>,
+        stack: &mut PodStack,
     ) {
         self.as_ref().transpose_apply(out, rhs, parallelism, stack)
     }
@@ -71,7 +71,7 @@ impl<E: ComplexField, ViewE: Conjugate<Canonical = E>> BiLinOp<E> for DiagMut<'_
         out: MatMut<'_, E>,
         rhs: MatRef<'_, E>,
         parallelism: Parallelism,
-        stack: PodStack<'_>,
+        stack: &mut PodStack,
     ) {
         self.as_ref().adjoint_apply(out, rhs, parallelism, stack)
     }
@@ -86,15 +86,15 @@ impl<E: ComplexField, ViewE: Conjugate<Canonical = E>> Precond<E> for DiagMut<'_
         self.as_ref().apply_in_place_req(rhs_ncols, parallelism)
     }
 
-    fn apply_in_place(&self, rhs: MatMut<'_, E>, parallelism: Parallelism, stack: PodStack<'_>) {
+    fn apply_in_place(&self, rhs: MatMut<'_, E>, parallelism: Parallelism, stack: &mut PodStack) {
         self.as_ref().apply_in_place(rhs, parallelism, stack)
     }
 
     fn conj_apply_in_place(
         &self,
         rhs: MatMut<'_, E>,
         parallelism: Parallelism,
-        stack: PodStack<'_>,
+        stack: &mut PodStack,
     ) {
         self.as_ref().conj_apply_in_place(rhs, parallelism, stack)
     }
@@ -113,7 +113,7 @@ impl<E: ComplexField, ViewE: Conjugate<Canonical = E>> BiPrecond<E> for DiagMut<
         &self,
         rhs: MatMut<'_, E>,
         parallelism: Parallelism,
-        stack: PodStack<'_>,
+        stack: &mut PodStack,
     ) {
         self.as_ref()
             .transpose_apply_in_place(rhs, parallelism, stack)
@@ -123,7 +123,7 @@ impl<E: ComplexField, ViewE: Conjugate<Canonical = E>> BiPrecond<E> for DiagMut<
         &self,
         rhs: MatMut<'_, E>,
         parallelism: Parallelism,
-        stack: PodStack<'_>,
+        stack: &mut PodStack,
     ) {
         self.as_ref()
             .adjoint_apply_in_place(rhs, parallelism, stack)