Program Listing for File linear_graph_partitioner.h

↰ Return to documentation for file (framework/graphs/linear_graph_partitioner.h)

// SPDX-FileCopyrightText: 2024 The OpenSn Authors <https://open-sn.github.io/opensn/>
// SPDX-License-Identifier: MIT

#pragma once

#include "framework/graphs/graph_partitioner.h"

namespace opensn
{

class LinearGraphPartitioner : public GraphPartitioner
{
public:
  explicit LinearGraphPartitioner(const InputParameters& params);

  std::vector<int64_t> Partition(const std::vector<std::vector<uint64_t>>& graph,
                                 const std::vector<Vector3>& centroids,
                                 int number_of_parts) override;

protected:
  const int all_to_rank_;

public:
  static InputParameters GetInputParameters();
  static std::shared_ptr<LinearGraphPartitioner> Create(const ParameterBlock& params);
};

} // namespace opensn