1*81ad6265SDimitry Andric //===- OffloadYAML.cpp - Offload Binary YAMLIO implementation -------------===//
2*81ad6265SDimitry Andric //
3*81ad6265SDimitry Andric // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4*81ad6265SDimitry Andric // See https://llvm.org/LICENSE.txt for license information.
5*81ad6265SDimitry Andric // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6*81ad6265SDimitry Andric //
7*81ad6265SDimitry Andric //===----------------------------------------------------------------------===//
8*81ad6265SDimitry Andric //
9*81ad6265SDimitry Andric // This file defines classes for handling the YAML representation of offload
10*81ad6265SDimitry Andric // binaries.
11*81ad6265SDimitry Andric //
12*81ad6265SDimitry Andric //===----------------------------------------------------------------------===//
13*81ad6265SDimitry Andric
14*81ad6265SDimitry Andric #include <llvm/ObjectYAML/OffloadYAML.h>
15*81ad6265SDimitry Andric
16*81ad6265SDimitry Andric namespace llvm {
17*81ad6265SDimitry Andric
18*81ad6265SDimitry Andric namespace yaml {
19*81ad6265SDimitry Andric
enumeration(IO & IO,object::ImageKind & Value)20*81ad6265SDimitry Andric void ScalarEnumerationTraits<object::ImageKind>::enumeration(
21*81ad6265SDimitry Andric IO &IO, object::ImageKind &Value) {
22*81ad6265SDimitry Andric #define ECase(X) IO.enumCase(Value, #X, object::X)
23*81ad6265SDimitry Andric ECase(IMG_None);
24*81ad6265SDimitry Andric ECase(IMG_Object);
25*81ad6265SDimitry Andric ECase(IMG_Bitcode);
26*81ad6265SDimitry Andric ECase(IMG_Cubin);
27*81ad6265SDimitry Andric ECase(IMG_Fatbinary);
28*81ad6265SDimitry Andric ECase(IMG_PTX);
29*81ad6265SDimitry Andric ECase(IMG_LAST);
30*81ad6265SDimitry Andric #undef ECase
31*81ad6265SDimitry Andric IO.enumFallback<Hex16>(Value);
32*81ad6265SDimitry Andric }
33*81ad6265SDimitry Andric
enumeration(IO & IO,object::OffloadKind & Value)34*81ad6265SDimitry Andric void ScalarEnumerationTraits<object::OffloadKind>::enumeration(
35*81ad6265SDimitry Andric IO &IO, object::OffloadKind &Value) {
36*81ad6265SDimitry Andric #define ECase(X) IO.enumCase(Value, #X, object::X)
37*81ad6265SDimitry Andric ECase(OFK_None);
38*81ad6265SDimitry Andric ECase(OFK_OpenMP);
39*81ad6265SDimitry Andric ECase(OFK_Cuda);
40*81ad6265SDimitry Andric ECase(OFK_HIP);
41*81ad6265SDimitry Andric ECase(OFK_LAST);
42*81ad6265SDimitry Andric #undef ECase
43*81ad6265SDimitry Andric IO.enumFallback<Hex16>(Value);
44*81ad6265SDimitry Andric }
45*81ad6265SDimitry Andric
mapping(IO & IO,OffloadYAML::Binary & O)46*81ad6265SDimitry Andric void MappingTraits<OffloadYAML::Binary>::mapping(IO &IO,
47*81ad6265SDimitry Andric OffloadYAML::Binary &O) {
48*81ad6265SDimitry Andric assert(!IO.getContext() && "The IO context is initialized already");
49*81ad6265SDimitry Andric IO.setContext(&O);
50*81ad6265SDimitry Andric IO.mapTag("!Offload", true);
51*81ad6265SDimitry Andric IO.mapOptional("Version", O.Version);
52*81ad6265SDimitry Andric IO.mapOptional("Size", O.Size);
53*81ad6265SDimitry Andric IO.mapOptional("EntryOffset", O.EntryOffset);
54*81ad6265SDimitry Andric IO.mapOptional("EntrySize", O.EntrySize);
55*81ad6265SDimitry Andric IO.mapRequired("Members", O.Members);
56*81ad6265SDimitry Andric IO.setContext(nullptr);
57*81ad6265SDimitry Andric }
58*81ad6265SDimitry Andric
mapping(IO & IO,OffloadYAML::Binary::StringEntry & SE)59*81ad6265SDimitry Andric void MappingTraits<OffloadYAML::Binary::StringEntry>::mapping(
60*81ad6265SDimitry Andric IO &IO, OffloadYAML::Binary::StringEntry &SE) {
61*81ad6265SDimitry Andric assert(IO.getContext() && "The IO context is not initialized");
62*81ad6265SDimitry Andric IO.mapRequired("Key", SE.Key);
63*81ad6265SDimitry Andric IO.mapRequired("Value", SE.Value);
64*81ad6265SDimitry Andric }
65*81ad6265SDimitry Andric
mapping(IO & IO,OffloadYAML::Binary::Member & M)66*81ad6265SDimitry Andric void MappingTraits<OffloadYAML::Binary::Member>::mapping(
67*81ad6265SDimitry Andric IO &IO, OffloadYAML::Binary::Member &M) {
68*81ad6265SDimitry Andric assert(IO.getContext() && "The IO context is not initialized");
69*81ad6265SDimitry Andric IO.mapOptional("ImageKind", M.ImageKind);
70*81ad6265SDimitry Andric IO.mapOptional("OffloadKind", M.OffloadKind);
71*81ad6265SDimitry Andric IO.mapOptional("Flags", M.Flags);
72*81ad6265SDimitry Andric IO.mapOptional("String", M.StringEntries);
73*81ad6265SDimitry Andric IO.mapOptional("Content", M.Content);
74*81ad6265SDimitry Andric }
75*81ad6265SDimitry Andric
76*81ad6265SDimitry Andric } // namespace yaml
77*81ad6265SDimitry Andric
78*81ad6265SDimitry Andric } // namespace llvm
79