Actual source code: htool.hpp

      1: #pragma once

      3: #include <petsc/private/matimpl.h>
      4: #include <petscmathtool.h>

      6: PETSC_PRAGMA_DIAGNOSTIC_IGNORED_BEGIN("-Wsign-compare")
      7: #include <htool/clustering/tree_builder/tree_builder.hpp>
      8: #include <htool/hmatrix/lrmat/SVD.hpp>
      9: #include <htool/hmatrix/lrmat/fullACA.hpp>
     10: #include <htool/hmatrix/lrmat/recompressed_low_rank_generator.hpp>
     11: #include <htool/hmatrix/hmatrix_distributed_output.hpp>
     12: #include <htool/hmatrix/linalg/factorization.hpp>
     13: #include <htool/distributed_operator/utility.hpp>
     14: #include <htool/distributed_operator/linalg/add_distributed_operator_vector_product_local_to_local.hpp>
     15: #include <htool/distributed_operator/linalg/add_distributed_operator_matrix_product_local_to_local.hpp>
     16: PETSC_PRAGMA_DIAGNOSTIC_IGNORED_END()

     18: class WrapperHtool : public htool::VirtualGenerator<PetscScalar> {
     19:   MatHtoolKernelFn *&kernel;
     20:   PetscInt           sdim;
     21:   void              *ctx;

     23: public:
     24:   WrapperHtool(PetscInt dim, MatHtoolKernelFn *&g, void *kernelctx) : VirtualGenerator<PetscScalar>(), kernel(g), sdim(dim), ctx(kernelctx) { }
     25:   void copy_submatrix(PetscInt M, PetscInt N, const PetscInt *rows, const PetscInt *cols, PetscScalar *ptr) const
     26:   {
     27: #if !PetscDefined(HAVE_OPENMP)
     28:     PetscFunctionBegin;
     29: #endif
     30:     PetscCallAbort(PETSC_COMM_SELF, kernel(sdim, M, N, rows, cols, ptr, ctx));
     31: #if !PetscDefined(HAVE_OPENMP)
     32:     PetscFunctionReturnVoid();
     33: #endif
     34:   }
     35: };

     37: struct Mat_Htool {
     38:   PetscInt                                                         dim;
     39:   PetscReal                                                       *gcoords_target;
     40:   PetscReal                                                       *gcoords_source;
     41:   PetscInt                                                         max_cluster_leaf_size;
     42:   PetscReal                                                        epsilon;
     43:   PetscReal                                                        eta;
     44:   PetscInt                                                         depth[2];
     45:   PetscBool                                                        block_tree_consistency;
     46:   PetscBool                                                        permutation;
     47:   PetscBool                                                        recompression;
     48:   MatHtoolCompressorType                                           compressor;
     49:   MatHtoolClusteringType                                           clustering;
     50:   MatHtoolKernelFn                                                *kernel;
     51:   void                                                            *kernelctx;
     52:   WrapperHtool                                                    *wrapper;
     53:   std::unique_ptr<htool::Cluster<PetscReal>>                       target_cluster;
     54:   std::unique_ptr<htool::Cluster<PetscReal>>                       source_cluster;
     55:   std::unique_ptr<htool::DefaultApproximationBuilder<PetscScalar>> distributed_operator_holder;
     56: };

     58: struct MatHtoolKernelTranspose {
     59:   Mat               A;
     60:   MatHtoolKernelFn *kernel;
     61:   void             *kernelctx;
     62: };