1 //=====-- NVPTXTargetStreamer.h - NVPTX Target Streamer ------*- C++ -*--=====//
2 //
3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4 // See https://llvm.org/LICENSE.txt for license information.
5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6 //
7 //===----------------------------------------------------------------------===//
8 
9 #ifndef LLVM_LIB_TARGET_NVPTX_MCTARGETDESC_NVPTXTARGETSTREAMER_H
10 #define LLVM_LIB_TARGET_NVPTX_MCTARGETDESC_NVPTXTARGETSTREAMER_H
11 
12 #include "llvm/MC/MCStreamer.h"
13 
14 namespace llvm {
15 class MCSection;
16 
17 /// Implments NVPTX-specific streamer.
18 class NVPTXTargetStreamer : public MCTargetStreamer {
19 private:
20   SmallVector<std::string, 4> DwarfFiles;
21   bool HasSections = false;
22 
23 public:
24   NVPTXTargetStreamer(MCStreamer &S);
25   ~NVPTXTargetStreamer() override;
26 
27   /// Outputs the list of the DWARF '.file' directives to the streamer.
28   void outputDwarfFileDirectives();
29   /// Close last section.
30   void closeLastSection();
31 
32   /// Record DWARF file directives for later output.
33   /// According to PTX ISA, CUDA Toolkit documentation, 11.5.3. Debugging
34   /// Directives: .file
35   /// (http://docs.nvidia.com/cuda/parallel-thread-execution/index.html#debugging-directives-file),
36   /// The .file directive is allowed only in the outermost scope, i.e., at the
37   /// same level as kernel and device function declarations. Also, the order of
38   /// the .loc and .file directive does not matter, .file directives may follow
39   /// the .loc directives where the file is referenced.
40   /// LLVM emits .file directives immediately the location debug info is
41   /// emitted, i.e. they may be emitted inside functions. We gather all these
42   /// directives and emit them outside of the sections and, thus, outside of the
43   /// functions.
44   void emitDwarfFileDirective(StringRef Directive) override;
45   void changeSection(const MCSection *CurSection, MCSection *Section,
46                      const MCExpr *SubSection, raw_ostream &OS) override;
47   /// Emit the bytes in \p Data into the output.
48   ///
49   /// This is used to emit bytes in \p Data as sequence of .byte directives.
50   void emitRawBytes(StringRef Data) override;
51 };
52 
53 class NVPTXAsmTargetStreamer : public NVPTXTargetStreamer {
54 public:
55   NVPTXAsmTargetStreamer(MCStreamer &S);
56   ~NVPTXAsmTargetStreamer() override;
57 };
58 
59 } // end namespace llvm
60 
61 #endif
62