File: ReducerWorkItem.h

package info (click to toggle)
llvm-toolchain-19 1%3A19.1.7-3
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid, trixie
  • size: 1,998,520 kB
  • sloc: cpp: 6,951,680; ansic: 1,486,157; asm: 913,598; python: 232,024; f90: 80,126; objc: 75,281; lisp: 37,276; pascal: 16,990; sh: 10,009; ml: 5,058; perl: 4,724; awk: 3,523; makefile: 3,167; javascript: 2,504; xml: 892; fortran: 664; cs: 573
file content (70 lines) | stat: -rw-r--r-- 2,157 bytes parent folder | download | duplicates (12)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
//===- ReducerWorkItem.h - Wrapper for Module -------------------*- C++ -*-===//
//
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
// See https://llvm.org/LICENSE.txt for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
//
//===----------------------------------------------------------------------===//

#ifndef LLVM_TOOLS_LLVM_REDUCE_REDUCERWORKITEM_H
#define LLVM_TOOLS_LLVM_REDUCE_REDUCERWORKITEM_H

#include "llvm/IR/Module.h"
#include <memory>

namespace llvm {
class LLVMContext;
class MachineModuleInfo;
class MemoryBufferRef;
class raw_ostream;
class TargetMachine;
class TestRunner;
struct BitcodeLTOInfo;

class ReducerWorkItem {
public:
  std::shared_ptr<Module> M;
  std::unique_ptr<BitcodeLTOInfo> LTOInfo;
  std::unique_ptr<MachineModuleInfo> MMI;

  ReducerWorkItem();
  ~ReducerWorkItem();
  ReducerWorkItem(ReducerWorkItem &) = delete;
  ReducerWorkItem(ReducerWorkItem &&) = default;

  bool isMIR() const { return MMI != nullptr; }

  LLVMContext &getContext() {
    return M->getContext();
  }

  Module &getModule() { return *M; }
  const Module &getModule() const { return *M; }
  operator Module &() const { return *M; }

  void print(raw_ostream &ROS, void *p = nullptr) const;
  bool verify(raw_fd_ostream *OS) const;
  std::unique_ptr<ReducerWorkItem> clone(const TargetMachine *TM) const;

  /// Return a number to indicate whether there was any reduction progress.
  uint64_t getComplexityScore() const {
    return isMIR() ? computeMIRComplexityScore() : computeIRComplexityScore();
  }

  void writeOutput(raw_ostream &OS, bool EmitBitcode) const;
  void readBitcode(MemoryBufferRef Data, LLVMContext &Ctx, StringRef ToolName);
  void writeBitcode(raw_ostream &OutStream) const;

  bool isReduced(const TestRunner &Test) const;

private:
  uint64_t computeIRComplexityScore() const;
  uint64_t computeMIRComplexityScore() const;
};

std::pair<std::unique_ptr<ReducerWorkItem>, bool>
parseReducerWorkItem(StringRef ToolName, StringRef Filename, LLVMContext &Ctxt,
                     std::unique_ptr<TargetMachine> &TM, bool IsMIR);
} // namespace llvm

#endif