diff options
Diffstat (limited to 'src/trace-driven/traces-generator/traces-post-processing/post-traces-processing.cpp')
| -rw-r--r-- | src/trace-driven/traces-generator/traces-post-processing/post-traces-processing.cpp | 195 |
1 files changed, 195 insertions, 0 deletions
diff --git a/src/trace-driven/traces-generator/traces-post-processing/post-traces-processing.cpp b/src/trace-driven/traces-generator/traces-post-processing/post-traces-processing.cpp new file mode 100644 index 0000000..0dbb4e1 --- /dev/null +++ b/src/trace-driven/traces-generator/traces-post-processing/post-traces-processing.cpp @@ -0,0 +1,195 @@ +#include <vector>
+#include <iostream>
+#include <fstream>
+#include <string>
+#include <sstream>
+#include <stdio.h>
+#include <math.h>
+using namespace std;
+
+struct threadblock_info
+{
+ bool initialized;
+ unsigned tb_id_x, tb_id_y, tb_id_z;
+ vector< vector< string > > warp_insts_array;
+ threadblock_info() {
+ initialized = false;
+ tb_id_x = tb_id_y = tb_id_z = 0;
+ }
+};
+
+void group_per_block(const char* filepath);
+void group_per_core(const char* filepath);
+
+//////////////////////////////////////////////////////////////////////////////////////////////////////////////////
+int main(int argc, char** argv)
+{
+
+ string kernellist_filepath;
+ bool is_per_core;
+ if(argc == 1)
+ {
+ cout << "File path is missing\n";
+ return 0;
+ } else if(argc == 2)
+ {
+ kernellist_filepath = argv[1];
+ is_per_core = true;
+
+ } else if(argc == 3) {
+ kernellist_filepath = argv[1];
+ is_per_core = bool(argv[2]);
+ }
+ else {
+ cout << "Too Many Arguemnts!\n";
+ return 0;
+ }
+
+ ifstream ifs;
+ ofstream ofs;
+
+ ifs.open(kernellist_filepath.c_str());
+ ofs.open((string(kernellist_filepath) + ".g").c_str());
+
+ if (!ifs.is_open()) {
+ cout << "Unable to open file: " <<kernellist_filepath<<endl;
+ return 0;
+ }
+
+
+ string directory(kernellist_filepath);
+ const size_t last_slash_idx = directory.rfind('/');
+ if (std::string::npos != last_slash_idx)
+ {
+ directory = directory.substr(0, last_slash_idx);
+ }
+
+ string line;
+ string filepath;
+ while(!ifs.eof()) {
+ getline(ifs, line);
+ if(line.empty())
+ continue;
+ filepath = directory+"/"+line;
+ group_per_block(filepath.c_str());
+ ofs<<line + "g"<<endl;
+ }
+
+ ifs.close();
+ ofs.close();
+ return 0;
+}
+
+
+void group_per_block(const char* filepath) {
+
+ ofstream ofs;
+ ifstream ifs;
+
+ ifs.open(filepath);
+
+ if (!ifs.is_open()) {
+ cout << "Unable to open file: " <<filepath<<endl;
+ return;
+ }
+
+ cout << "Processing file " <<filepath<<endl;
+ ofs.open((string(filepath) + "g").c_str());
+
+ vector<threadblock_info> insts;
+ unsigned grid_dim_x, grid_dim_y, grid_dim_z, tb_dim_x, tb_dim_y, tb_dim_z;
+ unsigned tb_id_x, tb_id_y, tb_id_z, tb_id, warpid_tb;
+ string line;
+ stringstream ss;
+ string string1, string2;
+ bool found_grid_dim = false, found_block_dim = false;
+
+ while(!ifs.eof()) {
+ getline(ifs, line);
+
+ if (line.length() == 0 || line[0] == '#') {
+ ofs<<line<<endl;
+ continue;
+ }
+
+ else if(line[0] == '-') {
+ ss.str(line);
+ ss.ignore();
+ ss>>string1>>string2;
+ if(string1 == "grid" && string2 == "dim") {
+ sscanf(line.c_str(), "-grid dim = (%d,%d,%d)", &grid_dim_x, &grid_dim_y, &grid_dim_z);
+ found_grid_dim = true;
+ }
+ else if (string1 == "block" && string2 == "dim") {
+ sscanf(line.c_str(), "-block dim = (%d,%d,%d)", &tb_dim_x, &tb_dim_y, &tb_dim_z);
+ found_block_dim = true;
+ }
+
+ if(found_grid_dim && found_block_dim) {
+ insts.resize(grid_dim_x*grid_dim_y*grid_dim_z);
+ for(unsigned i = 0; i<insts.size(); ++i) {
+ insts[i].warp_insts_array.resize(ceil(float(tb_dim_x*tb_dim_y*tb_dim_z)/32));
+ }
+ }
+ ofs<<line<<endl;
+ continue;
+ }
+ else {
+
+ ss.str(line);
+ ss>>tb_id_x>>tb_id_y>>tb_id_z>>warpid_tb;
+ tb_id = tb_id_z * grid_dim_y * grid_dim_x + tb_id_y * grid_dim_x + tb_id_x;
+ if(!insts[tb_id].initialized) {
+ insts[tb_id].tb_id_x = tb_id_x;
+ insts[tb_id].tb_id_y = tb_id_y;
+ insts[tb_id].tb_id_z = tb_id_z;
+ insts[tb_id].initialized = true;
+ }
+ insts[tb_id].warp_insts_array[warpid_tb].push_back(line);
+ }
+
+ }
+
+
+ for(unsigned i=0; i<insts.size(); ++i) {
+ //ofs<<string<<endl;
+ if(insts[i].initialized && insts[i].warp_insts_array.size() > 0) {
+ ofs<<endl<<"#BEGIN_TB"<<endl;
+ ofs<<endl<<"thread block = "<<insts[i].tb_id_x<<","<<insts[i].tb_id_y<<","<<insts[i].tb_id_z<<endl;
+ }
+ else {
+ cout<<"Warning: Thread block "<<insts[i].tb_id_x<<","<<insts[i].tb_id_y<<","<<insts[i].tb_id_z<<" is empty"<<endl;
+ continue;
+ //ofs.close();
+ //return;
+ }
+ for(unsigned j=0; j<insts[i].warp_insts_array.size(); ++j) {
+ ofs<<endl<<"warp = "<<j<<endl;
+ ofs<<"insts = "<<insts[i].warp_insts_array[j].size()<<endl;
+ if(insts[i].warp_insts_array[j].size() == 0) {
+ cout<<"Warning: Warp "<<j<<" in thread block"<<insts[i].tb_id_x<<","<<insts[i].tb_id_y<<","<<insts[i].tb_id_z<<" is empty"<<endl;
+ // ofs.close();
+ // return;
+ }
+ for(unsigned k=0; k<insts[i].warp_insts_array[j].size(); ++k) {
+ ofs<<insts[i].warp_insts_array[j][k]<<endl;
+ }
+ }
+ ofs<<endl<<"#END_TB"<<endl;
+ }
+
+
+
+ ofs.close();
+ ifs.close();
+
+}
+
+
+void group_per_core(const char* filepath)
+{
+
+ //TO DO
+
+
+}
|