aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--lib/Transforms/IPO/Cilkifier.cpp54
-rw-r--r--lib/Transforms/IPO/Cilkifier.h35
2 files changed, 89 insertions, 0 deletions
diff --git a/lib/Transforms/IPO/Cilkifier.cpp b/lib/Transforms/IPO/Cilkifier.cpp
new file mode 100644
index 0000000000..498c58b6fd
--- /dev/null
+++ b/lib/Transforms/IPO/Cilkifier.cpp
@@ -0,0 +1,54 @@
+//===- Cilkifier.cpp - Support routines for Cilk code gen. ------*- C++ -*-===//
+//
+// This is located here so that the code generator (dis) does not have to
+// include and link with the libtipo.a archive containing class Cilkifier
+// and the rest of the automatic parallelization code.
+//===----------------------------------------------------------------------===//
+
+
+#include "llvm/Support/Cilkifier.h"
+#include "llvm/Function.h"
+#include "llvm/iOther.h"
+#include "llvm/DerivedTypes.h"
+#include <vector>
+
+
+//----------------------------------------------------------------------------
+// Global constants used in marking Cilk functions and function calls.
+// These should be used only by the auto-parallelization pass.
+//----------------------------------------------------------------------------
+
+const std::string CilkSuffix(".llvm2cilk");
+const std::string DummySyncFuncName("__sync.llvm2cilk");
+
+//----------------------------------------------------------------------------
+// Routines to identify Cilk functions, calls to Cilk functions, and syncs.
+//----------------------------------------------------------------------------
+
+bool isCilk(const Function& F)
+{
+ assert(F.hasName());
+ return (F.getName().rfind(CilkSuffix) ==
+ F.getName().size() - CilkSuffix.size());
+}
+
+bool isCilkMain(const Function& F)
+{
+ assert(F.hasName());
+ return (F.getName() == std::string("main") + CilkSuffix);
+}
+
+
+bool isCilk(const CallInst& CI)
+{
+ return (CI.getCalledFunction() != NULL && isCilk(*CI.getCalledFunction()));
+}
+
+bool isSync(const CallInst& CI)
+{
+ return (CI.getCalledFunction() != NULL &&
+ CI.getCalledFunction()->getName() == DummySyncFuncName);
+}
+
+
+//----------------------------------------------------------------------------
diff --git a/lib/Transforms/IPO/Cilkifier.h b/lib/Transforms/IPO/Cilkifier.h
new file mode 100644
index 0000000000..be79ab6b22
--- /dev/null
+++ b/lib/Transforms/IPO/Cilkifier.h
@@ -0,0 +1,35 @@
+//===- Cilkifier.h - Support routines for Cilk code generation --*- C++ -*-===//
+//
+// This is located here so that the code generator (dis) does not have to
+// include and link with the libtipo.a archive containing class Cilkifier
+// and the rest of the automatic parallelization code.
+//===----------------------------------------------------------------------===//
+
+#ifndef LLVM_SUPPORT_CILKIFIER_H
+#define LLVM_SUPPORT_CILKIFIER_H
+
+#include <string>
+class Function;
+class CallInst;
+
+
+//----------------------------------------------------------------------------
+// Global constants used in marking Cilk functions and function calls.
+// These should be used only by the auto-parallelization pass.
+//----------------------------------------------------------------------------
+
+extern const std::string CilkSuffix;
+extern const std::string DummySyncFuncName;
+
+//----------------------------------------------------------------------------
+// Routines to identify Cilk functions, calls to Cilk functions, and syncs.
+//----------------------------------------------------------------------------
+
+extern bool isCilk (const Function& F);
+extern bool isCilkMain (const Function& F);
+extern bool isCilk (const CallInst& CI);
+extern bool isSync (const CallInst& CI);
+
+//===----------------------------------------------------------------------===//
+
+#endif