CuPBoP/compilation/KernelTranslation.cpp

63 lines
1.4 KiB
C++
Raw Normal View History

#include "generate_x86_format.h"
#include "handle_sync.h"
#include "init.h"
#include "insert_sync.h"
#include "insert_warp_loop.h"
#include "performance.h"
#include "tool.h"
#include "warp_func.h"
#include "llvm/IR/Module.h"
#include <assert.h>
2022-05-04 20:59:38 +08:00
#include <fstream>
#include <iostream>
2022-05-04 20:59:38 +08:00
#include <llvm/Support/raw_ostream.h>
#include <map>
#include <set>
#include <stdlib.h>
using namespace llvm;
2022-05-04 20:59:38 +08:00
std::string PATH = "kernel_meta.log";
int main(int argc, char **argv) {
2022-05-04 20:59:38 +08:00
assert(argc == 3 && "incorrect number of arguments\n");
llvm::Module *program = LoadModuleFromFilr(argv[1]);
2022-05-04 20:59:38 +08:00
std::ofstream fout;
fout.open(PATH);
// inline, and create auxiliary global variables
2022-05-04 20:59:38 +08:00
init_block(program, fout);
// insert sync before each vote, and replace the
// original vote function to warp vote
handle_warp_vote(program);
2022-05-04 20:59:38 +08:00
// replace warp shuffle
handle_warp_shfl(program);
2022-09-16 00:33:28 +08:00
// insert sync
insert_sync(program);
2022-09-16 00:33:28 +08:00
// split block by sync
split_block_by_sync(program);
// add loop for intra&intera thread
insert_warp_loop(program);
2022-05-04 20:59:38 +08:00
// (TODO): replace this patch
2022-05-04 20:59:38 +08:00
replace_built_in_function(program);
2022-09-16 00:33:28 +08:00
// TODO: replace with a more general function
// Not only for x86 backend
generate_x86_format(program);
2022-05-04 20:59:38 +08:00
// performance optimization
performance_optimization(program);
2022-05-04 20:59:38 +08:00
VerifyModule(program);
DumpModule(program, argv[2]);
2022-05-04 20:59:38 +08:00
fout.close();
return 0;
}