[llvm-commits] [poolalloc] r50353 - /poolalloc/trunk/lib/PoolAllocate/PASimple.cpp

[John Criswell]

Author: criswell
Date: Mon Apr 28 11:21:26 2008
New Revision: 50353

URL: http://llvm.org/viewvc/llvm-project?rev=50353&view=rev
Log:
Support FORTRAN programs converted by NAG's f2c program.

Modified:
    poolalloc/trunk/lib/PoolAllocate/PASimple.cpp

Modified: poolalloc/trunk/lib/PoolAllocate/PASimple.cpp
URL: http://llvm.org/viewvc/llvm-project/poolalloc/trunk/lib/PoolAllocate/PASimple.cpp?rev=50353&r1=50352&r2=50353&view=diff

==============================================================================
--- poolalloc/trunk/lib/PoolAllocate/PASimple.cpp (original)
+++ poolalloc/trunk/lib/PoolAllocate/PASimple.cpp Mon Apr 28 11:21:26 2008
@@ -89,7 +89,8 @@
   AddPoolPrototypes(&M);
 
   // Get the main function to insert the poolinit calls.
-  Function *MainFunc = M.getFunction("main");
+  Function *MainFunc = (M.getFunction("main") ? M.getFunction("main")
+                                              : M.getFunction("MAIN__"));
   if (MainFunc == 0 || MainFunc->isDeclaration()) {
     std::cerr << "Cannot pool allocate this program: it has global "
               << "pools but no 'main' function yet!\n";
@@ -104,7 +105,7 @@
   // Now that all call targets are available, rewrite the function bodies of the
   // clones.
   for (Module::iterator I = M.begin(), E = M.end(); I != E; ++I)
-    if (!I->isDeclaration())
+    if (!(I->isDeclaration()))
       ProcessFunctionBodySimple(*I);
   return true;
 }
@@ -333,7 +334,8 @@
                        Constant::getNullValue(getPoolType()), "GlobalPool",
                        &M);
 
-  Function *MainFunc = M.getFunction("main");
+  Function *MainFunc = (M.getFunction("main") ? M.getFunction("main")
+                                              : M.getFunction("MAIN__"));
   assert(MainFunc && "No main in program??");
 
   BasicBlock::iterator InsertPt;