GpuSparseLLT.h

// This file is part of Eigen, a lightweight C++ template library
// for linear algebra.
//
// Copyright (C) 2026 Rasmus Munk Larsen <rmlarsen@gmail.com>
//
// This Source Code Form is subject to the terms of the Mozilla
// Public License v. 2.0. If a copy of the MPL was not distributed
// with this file, You can obtain one at http://mozilla.org/MPL/2.0/.
// SPDX-License-Identifier: MPL-2.0
 
// GPU sparse Cholesky (LL^T / LL^H) via cuDSS.
 
#ifndef EIGEN_GPU_SPARSE_LLT_H
#define EIGEN_GPU_SPARSE_LLT_H
 
// IWYU pragma: private
#include "./InternalHeaderCheck.h"
 
#include "./GpuSparseSolverBase.h"
 
namespace Eigen {
namespace gpu {

template <typename Scalar_, int UpLo_ = Lower>
class SparseLLT : public internal::SparseSolverBase<Scalar_, SparseLLT<Scalar_, UpLo_>> {
  using Base = internal::SparseSolverBase<Scalar_, SparseLLT>;
  friend Base;
 
 public:
  using Scalar = Scalar_;
  static constexpr int UpLo = UpLo_;
 
  SparseLLT() = default;

  explicit SparseLLT(Context& ctx) : Base(ctx) {}
 
  template <typename InputType>
  explicit SparseLLT(const SparseMatrixBase<InputType>& A) {
    this->compute(A);
  }

  template <typename InputType>
  SparseLLT(Context& ctx, const SparseMatrixBase<InputType>& A) : Base(ctx) {
    this->compute(A);
  }
 
  static constexpr bool needs_csr_conversion() { return false; }
  static constexpr cudssMatrixType_t cudss_matrix_type() { return internal::cudss_spd_type<Scalar>::value; }
  static constexpr cudssMatrixViewType_t cudss_matrix_view() {
    return internal::cudss_view_type<UpLo, ColMajor>::value;
  }
};
 
}  // namespace gpu
}  // namespace Eigen
 
#endif  // EIGEN_GPU_SPARSE_LLT_H