colin
/
CuPBoP
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

update HostTranslator with debug tools

This commit is contained in:
Ruobing Han 2022-09-15 18:19:13 -04:00
parent bb3724c486
commit f2a4f7fe64
17 changed files with 37 additions and 32 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
common/debug.hpp Normal file
View File

@ -0,0 +1,14 @@
// This file is used for defining debug tools
#ifndef __DEBUG_TOOL__
#define __DEBUG_TOOL__
#include <stdio.h>
#ifdef DEBUG
#define DEBUG_INFO(...) fprintf(stderr, __VA_ARGS__)
#else
#define DEBUG_INFO(...)
#endif // DEBUG
#endif

1
compilation/HostTranslation/CMakeLists.txt
View File

@ -13,6 +13,7 @@ set(LIB_NAME cudaRuntime2cpuRuntime)
set(CMAKE_CXX_STANDARD 14)
set(CMAKE_BUILD_TYPE Debug)
include_directories(./include/x86)
include_directories(../../common)
file(GLOB proj_HEADERS "include/x86/*.h")
file(GLOB proj_SOURCES "src/x86/*.cpp")

7
compilation/HostTranslation/src/x86/GenerateHostStub.cpp
View File

@ -1,7 +0,0 @@
/**
* Generate a file for Cuda Kernel Function Attributes
*
*
*
*
*/

6
compilation/HostTranslation/src/x86/InitializeDevice.cpp
View File

@ -1,6 +0,0 @@
/*
Initialize the cudaDevice as first statements if not set by the User
(cudaSetDevice)
*/

4
compilation/HostTranslation/src/x86/RemoveCudaBuiltin.cpp
View File

@ -2,6 +2,7 @@
* Remove Clang cuda builtin functions and variables
*/
#include "RemoveCudaBuiltin.h"
#include "debug.hpp"
#include "llvm/IR/Function.h"
#include "llvm/IR/GlobalValue.h"
#include "llvm/IR/IRBuilder.h"
@ -46,8 +47,7 @@ void RemoveCudaBuiltin(llvm::Module *M) {
continue;
if (Ctors[i]->hasName() &&
Ctors[i]->getName().str().find("__cuda") == std::string::npos) {
std::cout << "keep: " << Ctors[i]->getName().str() << std::endl
<< std::flush;
DEBUG_INFO("keep: %s\n", Ctors[i]->getName().str().c_str());
CAList.push_back(OldCA->getOperand(i));
}
}

21
compilation/HostTranslation/src/x86/ReplaceCudaBuiltin.cpp
View File

@ -1,4 +1,5 @@
#include "ReplaceCudaBuiltin.h"
#include "debug.hpp"
#include "llvm/IR/Function.h"
#include "llvm/IR/GlobalValue.h"
#include "llvm/IR/IRBuilder.h"
@ -123,8 +124,8 @@ void ReplaceKernelLaunch(llvm::Module *M) {
cuda_register_kernel_names.insert(
functionOperand->getName().str());
std::cout << "Cuda Register Global Kernel: "
<< functionOperand->getName().str() << std::endl;
DEBUG_INFO("Cuda Register Global Kernel: %s\n",
functionOperand->getName().str().c_str());
}
}
}
@ -161,10 +162,10 @@ void ReplaceKernelLaunch(llvm::Module *M) {
dyn_cast<Function>(callOperand->stripPointerCasts());
FunctionType *ft = calledFunction->getFunctionType();
std::cout << " Parent (Caller) Function Name: " << func_name
<< ", cudaLaunchKernel Function: "
<< functionOperand->getName().str() << ", args "
<< functionOperand->arg_size() << std::endl;
DEBUG_INFO("Parent (Caller) Function Name: %s, "
"cudaLaunchKernel Function: %s, args : %d\n",
func_name, functionOperand->getName().str().c_str(),
functionOperand->arg_size());
auto rep = kernels.find(functionOperand->getName().str());
if (rep != kernels.end()) {
Function *FC = rep->second;
@ -205,8 +206,7 @@ void ReplaceKernelLaunch(llvm::Module *M) {
newName = newName.substr(i);
break;
}
std::cout << "Change Kernel Name to: " << newName << std::endl;
DEBUG_INFO("Change Kernel Name to: %s\n", newName.c_str());
Function *F =
Function::Create(FT, Function::ExternalLinkage, newName, M);
@ -225,9 +225,8 @@ void ReplaceKernelLaunch(llvm::Module *M) {
// for both cudaKernelLaunch calls and regular function call
host_changed = true;
calledFunction->setName(calledFunction->getName() + "_host");
std::cout << std::endl;
std::cout << "Change Host Function Name To: "
<< calledFunction->getName().str() << std::endl;
DEBUG_INFO("Change Host Function Name to: %s\n",
calledFunction->getName().str().c_str());
}
}
}

1
compilation/KernelTranslation/CMakeLists.txt
View File

@ -13,6 +13,7 @@ set(LIB_NAME spmd2mpmd)
set(CMAKE_CXX_STANDARD 14)
set(CMAKE_BUILD_TYPE Debug)
include_directories(./include/x86)
include_directories(../../common)
file(GLOB proj_HEADERS "include/x86/*.h")
file(GLOB proj_SOURCES "src/x86/*.cpp")

6
compilation/KernelTranslation/include/x86/tool.h
View File

@ -5,12 +5,6 @@
#include "llvm/IR/Instructions.h"
#include "llvm/IR/Module.h"
#ifdef DEBUG
#define DEBUG_INFO(...) fprintf(stderr, __VA_ARGS__)
#else
#define DEBUG_INFO(...)
#endif // DEBUG
llvm::Module *LoadModuleFromFilr(char *file_name);
void DumpModule(llvm::Module *M, char *file_name);
bool isKernelFunction(llvm::Module *M, llvm::Function *F);

1
compilation/KernelTranslation/src/x86/generate_x86_format.cpp
View File

@ -1,4 +1,5 @@
#include "generate_x86_format.h"
#include "debug.hpp"
#include "tool.h"
#include "llvm/Analysis/TargetLibraryInfo.h"
#include "llvm/Analysis/TargetTransformInfo.h"

1
compilation/KernelTranslation/src/x86/handle_sync.cpp
View File

@ -1,4 +1,5 @@
#include "handle_sync.h"
#include "debug.hpp"
#include "tool.h"
#include "llvm/IR/Function.h"
#include "llvm/IR/GlobalValue.h"

1
compilation/KernelTranslation/src/x86/init.cpp
View File

@ -1,4 +1,5 @@
#include "init.h"
#include "debug.hpp"
#include "memory_hierarchy.h"
#include "tool.h"
#include <fstream>

1
compilation/KernelTranslation/src/x86/insert_sync.cpp
View File

@ -1,5 +1,6 @@
#include "insert_sync.h"
#include "assert.h"
#include "debug.hpp"
#include "handle_sync.h"
#include "tool.h"
#include "llvm/ADT/SmallVector.h"

1
compilation/KernelTranslation/src/x86/insert_warp_loop.cpp
View File

@ -1,5 +1,6 @@
#include "insert_warp_loop.h"
#include "debug.hpp"
#include "handle_sync.h"
#include "tool.h"
#include <assert.h>

1
compilation/KernelTranslation/src/x86/memory_hierarchy.cpp
View File

@ -1,4 +1,5 @@
#include "memory_hierarchy.h"
#include "debug.hpp"
#include "llvm/IR/CFG.h"
#include "llvm/IR/Function.h"
#include "llvm/IR/GlobalValue.h"

1
compilation/KernelTranslation/src/x86/performance.cpp
View File

@ -1,4 +1,5 @@
#include "performance.h"
#include "debug.hpp"
#include "tool.h"
#include "llvm/ADT/Statistic.h"
#include "llvm/ADT/StringRef.h"

1
compilation/KernelTranslation/src/x86/tool.cpp
View File

@ -1,4 +1,5 @@
#include "tool.h"
#include "debug.hpp"
#include "llvm/Bitcode/BitcodeWriter.h"
#include "llvm/Config/llvm-config.h"
#include "llvm/IR/Constants.h"

1
compilation/KernelTranslation/src/x86/warp_func.cpp
View File

@ -1,5 +1,6 @@
#include "warp_func.h"
#include "debug.hpp"
#include "tool.h"
#include "llvm/IR/Function.h"
#include "llvm/IR/GlobalValue.h"