PTX: Add intrinsic support for ntid, ctaid, and nctaid registers

[oota-llvm.git] / lib / Target / PTX / PTX.h

diff --git a/lib/Target/PTX/PTX.h b/lib/Target/PTX/PTX.h
index 8e77e8d3711b4e9366847e77483dfd4e3acd9d20..19385ba1ff8c24ad9b9d52fc55ee86af874107d2 100644 (file)
--- a/lib/Target/PTX/PTX.h
+++ b/lib/Target/PTX/PTX.h
@@ -21,9 +21,22 @@ namespace llvm {
   class PTXTargetMachine;
   class FunctionPass;
 
+  namespace PTX {
+    enum StateSpace {
+      GLOBAL = 0, // default to global state space
+      CONSTANT = 1,
+      LOCAL = 2,
+      PARAMETER = 3,
+      SHARED = 4
+    };
+  } // namespace PTX
+
   FunctionPass *createPTXISelDag(PTXTargetMachine &TM,
                                  CodeGenOpt::Level OptLevel);
 
+  FunctionPass *createPTXMFInfoExtract(PTXTargetMachine &TM,
+                                       CodeGenOpt::Level OptLevel);
+
   extern Target ThePTXTarget;
 } // namespace llvm;