blob: 512c8e466acefafa0738d5495b500dbc5608e7dd (
plain) (
blame)
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
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
|
#pragma once
#ifdef __GNUC__
#pragma GCC diagnostic push
#pragma GCC diagnostic ignored "-Wunused-parameter"
#endif
//===- OffloadBundler.h - File Bundling and Unbundling ----------*- 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
//
//===----------------------------------------------------------------------===//
///
/// \file
/// This file defines an offload bundling API that bundles different files
/// that relate with the same source code but different targets into a single
/// one. Also the implements the opposite functionality, i.e. unbundle files
/// previous created by this API.
///
//===----------------------------------------------------------------------===//
#ifndef LLVM_CLANG_DRIVER_OFFLOADBUNDLER_H
#define LLVM_CLANG_DRIVER_OFFLOADBUNDLER_H
#include "llvm/ADT/Triple.h"
#include "llvm/Support/Error.h"
#include <string>
#include <vector>
namespace clang {
class OffloadBundlerConfig {
public:
bool AllowNoHost = false;
bool AllowMissingBundles = false;
bool CheckInputArchive = false;
bool PrintExternalCommands = false;
bool HipOpenmpCompatible = false;
unsigned BundleAlignment = 1;
unsigned HostInputIndex = ~0u;
std::string FilesType;
std::string ObjcopyPath;
// TODO: Convert these to llvm::SmallVector
std::vector<std::string> TargetNames;
std::vector<std::string> InputFileNames;
std::vector<std::string> OutputFileNames;
};
class OffloadBundler {
public:
const OffloadBundlerConfig &BundlerConfig;
// TODO: Add error checking from ClangOffloadBundler.cpp
OffloadBundler(const OffloadBundlerConfig &BC) : BundlerConfig(BC) {}
// List bundle IDs. Return true if an error was found.
static llvm::Error
ListBundleIDsInFile(llvm::StringRef InputFileName,
const OffloadBundlerConfig &BundlerConfig);
llvm::Error BundleFiles();
llvm::Error UnbundleFiles();
llvm::Error UnbundleArchive();
};
/// Obtain the offload kind, real machine triple, and an optional GPUArch
/// out of the target information specified by the user.
/// Bundle Entry ID (or, Offload Target String) has following components:
/// * Offload Kind - Host, OpenMP, or HIP
/// * Triple - Standard LLVM Triple
/// * TargetID (Optional) - target ID, like gfx906:xnack+ or sm_30
struct OffloadTargetInfo {
llvm::StringRef OffloadKind;
llvm::Triple Triple;
llvm::StringRef TargetID;
const OffloadBundlerConfig &BundlerConfig;
OffloadTargetInfo(const llvm::StringRef Target,
const OffloadBundlerConfig &BC);
bool hasHostKind() const;
bool isOffloadKindValid() const;
bool isOffloadKindCompatible(const llvm::StringRef TargetOffloadKind) const;
bool isTripleValid() const;
bool operator==(const OffloadTargetInfo &Target) const;
std::string str() const;
};
} // namespace clang
#endif // LLVM_CLANG_DRIVER_OFFLOADBUNDLER_H
#ifdef __GNUC__
#pragma GCC diagnostic pop
#endif
|
