86 typedef
Superlu<Matrix,Vector> type;
90 typedef typename super_type::scalar_type scalar_type;
91 typedef typename super_type::local_ordinal_type local_ordinal_type;
92 typedef typename super_type::global_ordinal_type global_ordinal_type;
93 typedef typename super_type::global_size_type global_size_type;
95 typedef
TypeMap<Amesos2::Superlu,scalar_type> type_map;
102 typedef typename type_map::type slu_type;
103 typedef typename type_map::convert_type slu_convert_type;
104 typedef typename type_map::magnitude_type magnitude_type;
106 typedef
FunctionMap<Amesos2::Superlu,slu_type> function_map;
117 Superlu(Teuchos::RCP<const Matrix> A,
118 Teuchos::RCP<Vector> X,
119 Teuchos::RCP<const Vector> B);
169 int
solve_impl(const Teuchos::Ptr<MultiVecAdapter<Vector> > X,
170 const Teuchos::Ptr<const MultiVecAdapter<Vector> > B) const;
213 const Teuchos::RCP<Teuchos::ParameterList> & parameterList );
235 typedef Kokkos::DefaultHostExecutionSpace HostExecSpaceType;
238 mutable struct SLUData {
239 SLU::SuperMatrix A, B, X, L, U;
242 SLU::superlu_options_t options;
243 SLU::mem_usage_t mem_usage;
244#ifdef HAVE_AMESOS2_SUPERLU5_API
247 SLU::SuperLUStat_t stat;
251 typedef Kokkos::View<magnitude_type*, HostExecSpaceType> host_mag_array;
252 typedef Kokkos::View<int*, HostExecSpaceType> host_int_array;
255 host_int_array perm_r;
256 host_int_array perm_c;
257 host_int_array etree;
261#if defined(KOKKOSKERNELS_ENABLE_SUPERNODAL_SPTRSV) && defined(KOKKOSKERNELS_ENABLE_TPL_SUPERLU)
262 host_int_array parents;
268 magnitude_type anorm, rcond;
274 typedef int size_type;
275 typedef int ordinal_type;
276 typedef Kokkos::View<size_type*, HostExecSpaceType> host_size_type_array;
277 typedef Kokkos::View<ordinal_type*, HostExecSpaceType> host_ordinal_type_array;
278 typedef Kokkos::View<slu_type*, HostExecSpaceType> host_value_type_array;
283 Teuchos::Array<slu_convert_type> convert_nzvals_;
290 typedef typename Kokkos::View<slu_type**, Kokkos::LayoutLeft, HostExecSpaceType>
295 mutable Teuchos::Array<slu_convert_type> convert_xValues_;
299 mutable Teuchos::Array<slu_convert_type> convert_bValues_;
301#if defined(KOKKOSKERNELS_ENABLE_SUPERNODAL_SPTRSV) && defined(KOKKOSKERNELS_ENABLE_TPL_SUPERLU)
302 typedef Kokkos::DefaultExecutionSpace DeviceExecSpaceType;
304 #ifdef KOKKOS_ENABLE_CUDA
306 typedef typename Kokkos::CudaSpace DeviceMemSpaceType;
308 typedef typename DeviceExecSpaceType::memory_space DeviceMemSpaceType;
311 typedef Kokkos::View<slu_type**, Kokkos::LayoutLeft, DeviceMemSpaceType>
312 device_solve_array_t;
315 mutable device_solve_array_t device_xValues_;
316 mutable device_solve_array_t device_bValues_;
317 typedef Kokkos::View<int*, DeviceMemSpaceType> device_int_array;
318 typedef Kokkos::View<magnitude_type*, DeviceMemSpaceType> device_mag_array;
319 device_int_array device_trsv_perm_r_;
320 device_int_array device_trsv_perm_c_;
321 device_mag_array device_trsv_R_;
322 device_mag_array device_trsv_C_;
323 mutable device_solve_array_t device_trsv_rhs_;
324 mutable device_solve_array_t device_trsv_sol_;
325 typedef KokkosKernels::Experimental::KokkosKernelsHandle <size_type, ordinal_type, slu_type,
326 DeviceExecSpaceType, DeviceMemSpaceType, DeviceMemSpaceType> kernel_handle_type;
327 mutable kernel_handle_type device_khL_;
328 mutable kernel_handle_type device_khU_;
330 bool sptrsv_invert_diag_;
331 bool sptrsv_invert_offdiag_;
332 bool sptrsv_u_in_csr_;
333 bool sptrsv_merge_supernodes_;
334 bool sptrsv_use_spmv_;
368 bool use_triangular_solves_;
370 void triangular_solve_factor();
374 bool symmetrize_metis_;
377 void triangular_solve()
const;