From f0c49462c773613155bb40febb75f807336edb3b Mon Sep 17 00:00:00 2001
From: Mahmoud <abdallm@purdue.edu>
Date: Tue, 24 Sep 2019 12:43:49 -0400
Subject: adding traces generator

---
 .../post-traces-processing.cpp                     | 195 +++++++++++++++++++++
 1 file changed, 195 insertions(+)
 create mode 100644 src/trace-driven/traces-generator/traces-post-processing/post-traces-processing.cpp

(limited to 'src/trace-driven/traces-generator/traces-post-processing/post-traces-processing.cpp')

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
+
+
+}
-- 
cgit v1.3