1 files changed, 54 insertions, 0 deletions
diff --git a/ir_cuda_suif_utils.cc b/ir_cuda_suif_utils.cc
new file mode 100644
index 0000000..f15c190
--- /dev/null
+++ b/ir_cuda_suif_utils.cc
@@ -0,0 +1,54 @@
+/*****************************************************************************
+ Copyright (C) 2008 University of Southern California
+ Copyright (C) 2009 University of Utah
+ All Rights Reserved.
+
+ Purpose:
+   SUIF interface utilities.
+
+ Notes:
+
+ Update history:
+   01/2006 created by Chun Chen
+*****************************************************************************/
+
+#include <suif1.h>
+#include "ir_suif_utils.hh"
+
+
+/**
+ * Returns the body of the for loop found by finding the first loop in
+ * code, and if level > 1 recursively calling on the body of the found
+ * loop and (level-1)
+ */
+tree_node_list* loop_body_at_level(tree_node_list* tnl, int level)
+{
+  tree_node_list *inner_nl = 0;
+  //Now strip out the tnl on the inner level of the for loop
+  tree_node_list_iter tnli(tnl);
+  while (!tnli.is_empty()) {
+    tree_node *node = tnli.step();
+    if(node->kind() == TREE_FOR)
+    {
+      //Found the first tree_for, call sibling function
+      inner_nl = loop_body_at_level((tree_for*)node, level);
+      break;
+    }
+  }
+  return inner_nl;
+}
+
+tree_node_list* loop_body_at_level(tree_for* loop, int level)
+{
+  if(level > 1)
+    return loop_body_at_level(loop->body(), level-1);
+  return loop->body();
+}
+
+tree_node_list*  swap_node_for_node_list(tree_node* tn, tree_node_list* new_tnl)
+{
+  tree_node_list* tnl  = tn->parent();
+  tnl->insert_after(new_tnl, tn->list_e());
+  delete tnl->remove(tn->list_e());
+  return tnl;
+}