FindGPUMacro.h

Go to the documentation of this file.

#ifndef __FIND_GPU_MACRO_H
#define __FIND_GPU_MACRO_H
 
#include "clang/AST/DeclCXX.h"
#include "clang/AST/RecursiveASTVisitor.h"
#include "llvm/Support/raw_ostream.h"
#include <map>
#include <set>
 
namespace systemc_clang {
 
using namespace std;
using namespace clang;
 
class GPUMacro {
 
public:
  GPUMacro(int, int, int, int, int, int, int, int);
  GPUMacro();
  ~GPUMacro();
 
  void addGPUFit();
  void denyGPUFit();
  int getBlockIdx();
  int getBlockIdy();
  int getBlockIdz();
  int getThreadIdx();
  int getThreadIdy();
  int getThreadIdz();
  int getGPUTime();
  int getCPUTime();
  bool isGPUFit();
  void dump(raw_ostream &);
 
private:
  int _blockIdx;
  int _blockIdy;
  int _blockIdz;
  int _threadIdx;
  int _threadIdy;
  int _threadIdz;
  int _gpuTime;
  int _cpuTime;
  bool _gpuFit;
};
 
class FindGPUMacro : public RecursiveASTVisitor<FindGPUMacro> {
public:
  typedef pair<int, ForStmt *> forStmtInstanceIdPairType;
  typedef map<int, ForStmt *> forStmtInstanceIdMapType;
 
  typedef pair<forStmtInstanceIdPairType, GPUMacro *> forStmtGPUMacroPairType;
  typedef map<forStmtInstanceIdPairType, GPUMacro *> forStmtGPUMacroMapType;
 
  FindGPUMacro(CXXMethodDecl *, int, raw_ostream &);
  virtual ~FindGPUMacro();
  // ANI : Need to add other loops as well.....
  virtual bool VisitForStmt(ForStmt *);
 
  forStmtGPUMacroMapType getForStmtGPUMacroMap();
 
  void dump();
 
  // ===
  // Analyze the simple for loop
  //
  // int a[5];
  // for (int i = 0; i < 5; ++i) {
  //     a[i] = i;
  // }
  // ===
  void analyze_data_struct(Stmt *stmtList);
  void analyze_expr(Expr *expr);
  void analyze_lhs(Expr *expr);
  void analyze_rhs(Expr *expr);
  void analyze_array_base(Expr *base, bool isLHS);
  void analyze_value_decl(ValueDecl *val);
 
private:
  forStmtGPUMacroMapType _forStmtGPUMacroMap;
  CXXMethodDecl *_entryFunction;
  int _instanceNum;
  raw_ostream &_os;
 
  std::set<ValueDecl *> lhs_decls;
  std::set<ValueDecl *> rhs_decls;
};
} // namespace systemc_clang
#endif