File tree Expand file tree Collapse file tree
DifferentiationInterface/ext/DifferentiationInterfaceSparseMatrixColoringsExt Expand file tree Collapse file tree Original file line number Diff line number Diff line change 139139function _prepare_sparse_jacobian_aux_aux (
140140 strict:: Val ,
141141 batch_size_settings:: DI.BatchSizeSettings{B} ,
142+ sparsity:: AbstractMatrix ,
142143 coloring_result:: AbstractColoringResult{:nonsymmetric,:row} ,
143144 y,
144145 f_or_f!y:: FY ,
@@ -162,6 +163,7 @@ function _prepare_sparse_jacobian_aux_aux(
162163 return SMCPullbackSparseJacobianPrep (
163164 _sig,
164165 batch_size_settings,
166+ sparsity,
165167 coloring_result,
166168 compressed_matrix,
167169 batched_seeds,
Original file line number Diff line number Diff line change @@ -94,6 +94,7 @@ function _prepare_mixed_sparse_jacobian_aux_aux(
9494 strict:: Val ,
9595 batch_size_settings_forward:: DI.BatchSizeSettings{Bf} ,
9696 batch_size_settings_reverse:: DI.BatchSizeSettings{Br} ,
97+ sparsity:: AbstractMatrix ,
9798 coloring_result:: AbstractColoringResult{:nonsymmetric,:bidirectional} ,
9899 y,
99100 f_or_f!y:: FY ,
@@ -151,6 +152,7 @@ function _prepare_mixed_sparse_jacobian_aux_aux(
151152 _sig,
152153 batch_size_settings_forward,
153154 batch_size_settings_reverse,
155+ sparsity,
154156 coloring_result,
155157 compressed_matrix_forward,
156158 compressed_matrix_reverse,
You can’t perform that action at this time.
0 commit comments