added working ptxc file
This commit is contained in:
Evghenii
File diff suppressed because it is too large
Load Diff
@@ -1,275 +0,0 @@
|
|||||||
#ifndef _ANYOPTION_H
|
|
||||||
#define _ANYOPTION_H
|
|
||||||
|
|
||||||
/* soruce: https://github.com/hackorama/AnyOption/ */
|
|
||||||
|
|
||||||
#include <iostream>
|
|
||||||
#include <fstream>
|
|
||||||
#include <cstdlib>
|
|
||||||
#include <vector>
|
|
||||||
#include <string>
|
|
||||||
#include <cstring>
|
|
||||||
|
|
||||||
#define COMMON_OPT 1
|
|
||||||
#define COMMAND_OPT 2
|
|
||||||
#define FILE_OPT 3
|
|
||||||
#define COMMON_FLAG 4
|
|
||||||
#define COMMAND_FLAG 5
|
|
||||||
#define FILE_FLAG 6
|
|
||||||
|
|
||||||
#define COMMAND_OPTION_TYPE 1
|
|
||||||
#define COMMAND_FLAG_TYPE 2
|
|
||||||
#define FILE_OPTION_TYPE 3
|
|
||||||
#define FILE_FLAG_TYPE 4
|
|
||||||
#define UNKNOWN_TYPE 5
|
|
||||||
|
|
||||||
#define DEFAULT_MAXOPTS 10
|
|
||||||
#define MAX_LONG_PREFIX_LENGTH 2
|
|
||||||
|
|
||||||
#define DEFAULT_MAXUSAGE 3
|
|
||||||
#define DEFAULT_MAXHELP 10
|
|
||||||
|
|
||||||
#define TRUE_FLAG "true"
|
|
||||||
|
|
||||||
|
|
||||||
class AnyOption
|
|
||||||
{
|
|
||||||
|
|
||||||
public: /* the public interface */
|
|
||||||
AnyOption();
|
|
||||||
AnyOption(int maxoptions );
|
|
||||||
AnyOption(int maxoptions , int maxcharoptions);
|
|
||||||
~AnyOption();
|
|
||||||
|
|
||||||
/*
|
|
||||||
* following set methods specifies the
|
|
||||||
* special characters and delimiters
|
|
||||||
* if not set traditional defaults will be used
|
|
||||||
*/
|
|
||||||
|
|
||||||
void setCommandPrefixChar( char _prefix ); /* '-' in "-w" */
|
|
||||||
void setCommandLongPrefix( char *_prefix ); /* '--' in "--width" */
|
|
||||||
void setFileCommentChar( char _comment ); /* '#' in shellscripts */
|
|
||||||
void setFileDelimiterChar( char _delimiter );/* ':' in "width : 100" */
|
|
||||||
|
|
||||||
/*
|
|
||||||
* provide the input for the options
|
|
||||||
* like argv[] for commndline and the
|
|
||||||
* option file name to use;
|
|
||||||
*/
|
|
||||||
|
|
||||||
void useCommandArgs( int _argc, char **_argv );
|
|
||||||
void useFiileName( const char *_filename );
|
|
||||||
|
|
||||||
/*
|
|
||||||
* turn off the POSIX style options
|
|
||||||
* this means anything starting with a '-' or "--"
|
|
||||||
* will be considered a valid option
|
|
||||||
* which alo means you cannot add a bunch of
|
|
||||||
* POIX options chars together like "-lr" for "-l -r"
|
|
||||||
*
|
|
||||||
*/
|
|
||||||
|
|
||||||
void noPOSIX();
|
|
||||||
|
|
||||||
/*
|
|
||||||
* prints warning verbose if you set anything wrong
|
|
||||||
*/
|
|
||||||
void setVerbose();
|
|
||||||
|
|
||||||
|
|
||||||
/*
|
|
||||||
* there are two types of options
|
|
||||||
*
|
|
||||||
* Option - has an associated value ( -w 100 )
|
|
||||||
* Flag - no value, just a boolean flag ( -nogui )
|
|
||||||
*
|
|
||||||
* the options can be either a string ( GNU style )
|
|
||||||
* or a character ( traditional POSIX style )
|
|
||||||
* or both ( --width, -w )
|
|
||||||
*
|
|
||||||
* the options can be common to the commandline and
|
|
||||||
* the optionfile, or can belong only to either of
|
|
||||||
* commandline and optionfile
|
|
||||||
*
|
|
||||||
* following set methods, handle all the aboove
|
|
||||||
* cases of options.
|
|
||||||
*/
|
|
||||||
|
|
||||||
/* options comman to command line and option file */
|
|
||||||
void setOption( const char *opt_string );
|
|
||||||
void setOption( char opt_char );
|
|
||||||
void setOption( const char *opt_string , char opt_char );
|
|
||||||
void setFlag( const char *opt_string );
|
|
||||||
void setFlag( char opt_char );
|
|
||||||
void setFlag( const char *opt_string , char opt_char );
|
|
||||||
|
|
||||||
/* options read from commandline only */
|
|
||||||
void setCommandOption( const char *opt_string );
|
|
||||||
void setCommandOption( char opt_char );
|
|
||||||
void setCommandOption( const char *opt_string , char opt_char );
|
|
||||||
void setCommandFlag( const char *opt_string );
|
|
||||||
void setCommandFlag( char opt_char );
|
|
||||||
void setCommandFlag( const char *opt_string , char opt_char );
|
|
||||||
|
|
||||||
/* options read from an option file only */
|
|
||||||
void setFileOption( const char *opt_string );
|
|
||||||
void setFileOption( char opt_char );
|
|
||||||
void setFileOption( const char *opt_string , char opt_char );
|
|
||||||
void setFileFlag( const char *opt_string );
|
|
||||||
void setFileFlag( char opt_char );
|
|
||||||
void setFileFlag( const char *opt_string , char opt_char );
|
|
||||||
|
|
||||||
/*
|
|
||||||
* process the options, registerd using
|
|
||||||
* useCommandArgs() and useFileName();
|
|
||||||
*/
|
|
||||||
void processOptions();
|
|
||||||
void processCommandArgs();
|
|
||||||
void processCommandArgs( int max_args );
|
|
||||||
bool processFile();
|
|
||||||
|
|
||||||
/*
|
|
||||||
* process the specified options
|
|
||||||
*/
|
|
||||||
void processCommandArgs( int _argc, char **_argv );
|
|
||||||
void processCommandArgs( int _argc, char **_argv, int max_args );
|
|
||||||
bool processFile( const char *_filename );
|
|
||||||
|
|
||||||
/*
|
|
||||||
* get the value of the options
|
|
||||||
* will return NULL if no value is set
|
|
||||||
*/
|
|
||||||
char *getValue( const char *_option );
|
|
||||||
bool getFlag( const char *_option );
|
|
||||||
char *getValue( char _optchar );
|
|
||||||
bool getFlag( char _optchar );
|
|
||||||
|
|
||||||
/*
|
|
||||||
* Print Usage
|
|
||||||
*/
|
|
||||||
void printUsage();
|
|
||||||
void printAutoUsage();
|
|
||||||
void addUsage( const char *line );
|
|
||||||
void addUsage( const std::string &line);
|
|
||||||
void printHelp();
|
|
||||||
/* print auto usage printing for unknown options or flag */
|
|
||||||
void autoUsagePrint(bool flag);
|
|
||||||
|
|
||||||
/*
|
|
||||||
* get the argument count and arguments sans the options
|
|
||||||
*/
|
|
||||||
int getArgc();
|
|
||||||
char* getArgv( int index );
|
|
||||||
bool hasOptions();
|
|
||||||
|
|
||||||
private: /* the hidden data structure */
|
|
||||||
int argc; /* commandline arg count */
|
|
||||||
char **argv; /* commndline args */
|
|
||||||
const char* filename; /* the option file */
|
|
||||||
char* appname; /* the application name from argv[0] */
|
|
||||||
|
|
||||||
int *new_argv; /* arguments sans options (index to argv) */
|
|
||||||
int new_argc; /* argument count sans the options */
|
|
||||||
int max_legal_args; /* ignore extra arguments */
|
|
||||||
|
|
||||||
|
|
||||||
/* option strings storage + indexing */
|
|
||||||
int max_options; /* maximum number of options */
|
|
||||||
const char **options; /* storage */
|
|
||||||
int *optiontype; /* type - common, command, file */
|
|
||||||
int *optionindex; /* index into value storage */
|
|
||||||
int option_counter; /* counter for added options */
|
|
||||||
|
|
||||||
/* option chars storage + indexing */
|
|
||||||
int max_char_options; /* maximum number options */
|
|
||||||
char *optionchars; /* storage */
|
|
||||||
int *optchartype; /* type - common, command, file */
|
|
||||||
int *optcharindex; /* index into value storage */
|
|
||||||
int optchar_counter; /* counter for added options */
|
|
||||||
|
|
||||||
/* values */
|
|
||||||
char **values; /* common value storage */
|
|
||||||
int g_value_counter; /* globally updated value index LAME! */
|
|
||||||
|
|
||||||
/* help and usage */
|
|
||||||
const char **usage; /* usage */
|
|
||||||
std::vector<std::string> usage_string;
|
|
||||||
int max_usage_lines; /* max usage lines reseverd */
|
|
||||||
int usage_lines; /* number of usage lines */
|
|
||||||
|
|
||||||
bool command_set; /* if argc/argv were provided */
|
|
||||||
bool file_set; /* if a filename was provided */
|
|
||||||
bool mem_allocated; /* if memory allocated in init() */
|
|
||||||
bool posix_style; /* enables to turn off POSIX style options */
|
|
||||||
bool verbose; /* silent|verbose */
|
|
||||||
bool print_usage; /* usage verbose */
|
|
||||||
bool print_help; /* help verbose */
|
|
||||||
|
|
||||||
char opt_prefix_char; /* '-' in "-w" */
|
|
||||||
char long_opt_prefix[MAX_LONG_PREFIX_LENGTH + 1]; /* '--' in "--width" */
|
|
||||||
char file_delimiter_char; /* ':' in width : 100 */
|
|
||||||
char file_comment_char; /* '#' in "#this is a comment" */
|
|
||||||
char equalsign;
|
|
||||||
char comment;
|
|
||||||
char delimiter;
|
|
||||||
char endofline;
|
|
||||||
char whitespace;
|
|
||||||
char nullterminate;
|
|
||||||
|
|
||||||
bool set; //was static member
|
|
||||||
bool once; //was static member
|
|
||||||
|
|
||||||
bool hasoptions;
|
|
||||||
bool autousage;
|
|
||||||
|
|
||||||
private: /* the hidden utils */
|
|
||||||
void init();
|
|
||||||
void init(int maxopt, int maxcharopt );
|
|
||||||
bool alloc();
|
|
||||||
void cleanup();
|
|
||||||
bool valueStoreOK();
|
|
||||||
|
|
||||||
/* grow storage arrays as required */
|
|
||||||
bool doubleOptStorage();
|
|
||||||
bool doubleCharStorage();
|
|
||||||
bool doubleUsageStorage();
|
|
||||||
|
|
||||||
bool setValue( const char *option , char *value );
|
|
||||||
bool setFlagOn( const char *option );
|
|
||||||
bool setValue( char optchar , char *value);
|
|
||||||
bool setFlagOn( char optchar );
|
|
||||||
|
|
||||||
void addOption( const char* option , int type );
|
|
||||||
void addOption( char optchar , int type );
|
|
||||||
void addOptionError( const char *opt);
|
|
||||||
void addOptionError( char opt);
|
|
||||||
bool findFlag( char* value );
|
|
||||||
void addUsageError( const char *line );
|
|
||||||
bool CommandSet();
|
|
||||||
bool FileSet();
|
|
||||||
bool POSIX();
|
|
||||||
|
|
||||||
char parsePOSIX( char* arg );
|
|
||||||
int parseGNU( char *arg );
|
|
||||||
bool matchChar( char c );
|
|
||||||
int matchOpt( char *opt );
|
|
||||||
|
|
||||||
/* dot file methods */
|
|
||||||
char *readFile();
|
|
||||||
char *readFile( const char* fname );
|
|
||||||
bool consumeFile( char *buffer );
|
|
||||||
void processLine( char *theline, int length );
|
|
||||||
char *chomp( char *str );
|
|
||||||
void valuePairs( char *type, char *value );
|
|
||||||
void justValue( char *value );
|
|
||||||
|
|
||||||
void printVerbose( const char *msg );
|
|
||||||
void printVerbose( char *msg );
|
|
||||||
void printVerbose( char ch );
|
|
||||||
void printVerbose( );
|
|
||||||
|
|
||||||
|
|
||||||
};
|
|
||||||
|
|
||||||
#endif /* ! _ANYOPTION_H */
|
|
||||||
228
examples_ptx/ptxcc/ptxc.cpp
Normal file
228
examples_ptx/ptxcc/ptxc.cpp
Normal file
@@ -0,0 +1,228 @@
|
|||||||
|
#include <cstdio>
|
||||||
|
#include <cstdlib>
|
||||||
|
#include <iostream>
|
||||||
|
#include <fstream>
|
||||||
|
#include <cassert>
|
||||||
|
#include <algorithm>
|
||||||
|
#include "PTXParser.h"
|
||||||
|
|
||||||
|
static char lRandomChar()
|
||||||
|
{
|
||||||
|
const char charset[] =
|
||||||
|
"0123456789"
|
||||||
|
"ABCDEFGHIJKLMNOPQRSTUVWXYZ"
|
||||||
|
"abcdefghijklmnopqrstuvwxyz";
|
||||||
|
const size_t max_index = (sizeof(charset) - 1);
|
||||||
|
return charset[ rand() % max_index ];
|
||||||
|
}
|
||||||
|
|
||||||
|
static std::string lRandomString(const size_t length)
|
||||||
|
{
|
||||||
|
std::string str(length,0);
|
||||||
|
std::generate_n( str.begin(), length, lRandomChar);
|
||||||
|
return str;
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
static void lGetAllArgs(int Argc, char *Argv[], int &argc, char *argv[128])
|
||||||
|
{
|
||||||
|
// Copy over the command line arguments (passed in)
|
||||||
|
for (int i = 0; i < Argc; ++i)
|
||||||
|
argv[i] = Argv[i];
|
||||||
|
argc = Argc;
|
||||||
|
}
|
||||||
|
const char *lGetExt (const char *fspec)
|
||||||
|
{
|
||||||
|
char *e = strrchr (fspec, '.');
|
||||||
|
return e;
|
||||||
|
}
|
||||||
|
|
||||||
|
static std::vector<std::string> lSplitString(const std::string &s, char delim)
|
||||||
|
{
|
||||||
|
std::vector<std::string> elems;
|
||||||
|
std::stringstream ss(s);
|
||||||
|
std::string item;
|
||||||
|
while (std::getline(ss, item, delim)) {
|
||||||
|
if (!item.empty())
|
||||||
|
elems.push_back(item);
|
||||||
|
}
|
||||||
|
return elems;
|
||||||
|
}
|
||||||
|
|
||||||
|
static void usage(const int ret)
|
||||||
|
{
|
||||||
|
exit(ret);
|
||||||
|
}
|
||||||
|
|
||||||
|
int main(int _argc, char * _argv[])
|
||||||
|
{
|
||||||
|
int argc;
|
||||||
|
char *argv[128];
|
||||||
|
lGetAllArgs(_argc, _argv, argc, argv);
|
||||||
|
|
||||||
|
std::string arch="sm_35";
|
||||||
|
std::string filePTX;
|
||||||
|
std::string fileOBJ;
|
||||||
|
std::string extString = ".ptx";
|
||||||
|
bool keepTemporaries = false;
|
||||||
|
|
||||||
|
std::vector<std::string> nvccArgumentList;
|
||||||
|
|
||||||
|
for (int i = 1; i < argc; ++i)
|
||||||
|
{
|
||||||
|
if (!strcmp(argv[i], "--help"))
|
||||||
|
usage(0);
|
||||||
|
else if (!strncmp(argv[i], "--arch=", 7))
|
||||||
|
arch = std::string(argv[i]+7);
|
||||||
|
else if (!strncmp(argv[i], "--keep-temporaries", 11))
|
||||||
|
keepTemporaries = true;
|
||||||
|
else if (!strcmp(argv[i], "-o"))
|
||||||
|
{
|
||||||
|
if (++i == argc)
|
||||||
|
{
|
||||||
|
fprintf(stderr, "No output file specified after -o option.\n");
|
||||||
|
usage(1);
|
||||||
|
}
|
||||||
|
fileOBJ = std::string(argv[i]);
|
||||||
|
}
|
||||||
|
else if (strncmp(argv[i], "-", 1))
|
||||||
|
{
|
||||||
|
const char * ext = strrchr(argv[i], '.');
|
||||||
|
if (ext == NULL)
|
||||||
|
{
|
||||||
|
fprintf(stderr, " Unknown argument: %s \n", argv[i]);
|
||||||
|
exit(1);
|
||||||
|
}
|
||||||
|
else if (strncmp(ext, extString.c_str(), 4))
|
||||||
|
{
|
||||||
|
fprintf(stderr, " Unkown extension of the input file: %s \n", ext);
|
||||||
|
exit(1);
|
||||||
|
}
|
||||||
|
else if (filePTX.empty())
|
||||||
|
{
|
||||||
|
filePTX = std::string(argv[i]);
|
||||||
|
if (fileOBJ.empty())
|
||||||
|
{
|
||||||
|
char * baseName = argv[i];
|
||||||
|
while (baseName != ext)
|
||||||
|
fileOBJ += std::string(baseName++,1);
|
||||||
|
}
|
||||||
|
fileOBJ += ".o";
|
||||||
|
}
|
||||||
|
}
|
||||||
|
else
|
||||||
|
nvccArgumentList.push_back(argv[i]);
|
||||||
|
}
|
||||||
|
#if 0
|
||||||
|
fprintf(stderr, " fileOBJ= %s\n", fileOBJ.c_str());
|
||||||
|
fprintf(stderr, " arch= %s\n", arch.c_str());
|
||||||
|
fprintf(stderr, " file= %s\n", filePTX.empty() ? "$stdin" : filePTX.c_str());
|
||||||
|
fprintf(stderr, " num_args= %d\n", (int)nvccArgumentList.size());
|
||||||
|
for (int i= 0; i < (int)nvccArgumentList.size(); i++)
|
||||||
|
fprintf(stderr, " arg= %d : %s \n", i, nvccArgumentList[i].c_str());
|
||||||
|
#endif
|
||||||
|
assert(arch == std::string("sm_35"));
|
||||||
|
if (filePTX.empty())
|
||||||
|
{
|
||||||
|
fprintf(stderr, "ptxc fatal : No input file specified; use option --help for more information\n");
|
||||||
|
exit(1);
|
||||||
|
}
|
||||||
|
|
||||||
|
// open a file handle to a particular file:
|
||||||
|
std::ifstream inputPTX(filePTX.c_str());
|
||||||
|
if (!inputPTX)
|
||||||
|
{
|
||||||
|
fprintf(stderr, "ptxc: error: %s: No such file\n", filePTX.c_str());
|
||||||
|
exit(1);
|
||||||
|
}
|
||||||
|
|
||||||
|
std::string randomBaseName = std::string("/tmp/") + lRandomString(12);
|
||||||
|
fprintf(stderr, " randomBaseName= %s\n", randomBaseName.c_str());
|
||||||
|
|
||||||
|
std::string fileCU= randomBaseName + ".cu";
|
||||||
|
std::ofstream outputCU(fileCU.c_str());
|
||||||
|
assert(outputCU);
|
||||||
|
|
||||||
|
std::istream & input = inputPTX;
|
||||||
|
std::ostream & output = outputCU;
|
||||||
|
std::ostream & error = std::cerr;
|
||||||
|
parser::PTXLexer lexer(&input, &error);
|
||||||
|
parser::PTXParser state(output);
|
||||||
|
|
||||||
|
// parse through the input until there is no more:
|
||||||
|
//
|
||||||
|
|
||||||
|
do {
|
||||||
|
ptx::yyparse(lexer, state);
|
||||||
|
}
|
||||||
|
while (!input.eof());
|
||||||
|
|
||||||
|
inputPTX.close();
|
||||||
|
outputCU.close();
|
||||||
|
|
||||||
|
// process output from nvcc
|
||||||
|
//
|
||||||
|
/* nvcc -dc -arch=$arch -dryrun -argumentlist fileCU */
|
||||||
|
|
||||||
|
std::string fileSH= randomBaseName + ".sh";
|
||||||
|
|
||||||
|
std::string nvccExe("nvcc");
|
||||||
|
std::string nvccCmd;
|
||||||
|
nvccCmd += nvccExe + std::string(" ");
|
||||||
|
nvccCmd += "-dc ";
|
||||||
|
nvccCmd += std::string("-arch=") + arch + std::string(" ");
|
||||||
|
nvccCmd += "-dryrun ";
|
||||||
|
for (int i = 0; i < (int)nvccArgumentList.size(); i++)
|
||||||
|
nvccCmd += nvccArgumentList[i] + std::string(" ");
|
||||||
|
nvccCmd += std::string("-o ") + fileOBJ + std::string(" ");
|
||||||
|
nvccCmd += fileCU + std::string(" ");
|
||||||
|
nvccCmd += std::string("2> ") + fileSH;
|
||||||
|
fprintf(stderr , " nvccCmd= %s\n", nvccCmd.c_str());
|
||||||
|
std::system(nvccCmd.c_str());
|
||||||
|
|
||||||
|
|
||||||
|
std::ifstream inputSH(fileSH.c_str());
|
||||||
|
assert(inputSH);
|
||||||
|
std::vector<std::string> nvccSteps;
|
||||||
|
while (!inputSH.eof())
|
||||||
|
{
|
||||||
|
nvccSteps.push_back(std::string());
|
||||||
|
std::getline(inputSH, nvccSteps.back());
|
||||||
|
}
|
||||||
|
inputSH.close();
|
||||||
|
|
||||||
|
for (int i = 0; i < (int)nvccSteps.size(); i++)
|
||||||
|
{
|
||||||
|
std::string cmd = nvccSteps[i];
|
||||||
|
for (int j = 0; j < (int)cmd.size()-1; j++)
|
||||||
|
if (cmd[j] == '#' && cmd[j+1] == '$')
|
||||||
|
cmd[j] = cmd[j+1] = ' ';
|
||||||
|
std::vector<std::string> splitCmd = lSplitString(cmd, ' ');
|
||||||
|
|
||||||
|
if (!splitCmd.empty())
|
||||||
|
{
|
||||||
|
if (splitCmd[0] == std::string("cicc"))
|
||||||
|
cmd = std::string("cp ") + filePTX + std::string(" ") + splitCmd.back();
|
||||||
|
if (splitCmd[0] == std::string("LIBRARIES="))
|
||||||
|
cmd = "";
|
||||||
|
}
|
||||||
|
nvccSteps[i] = cmd;
|
||||||
|
const int ret = std::system(cmd.c_str());
|
||||||
|
if (ret)
|
||||||
|
{
|
||||||
|
fprintf(stderr, " Something went wrong .. \n");
|
||||||
|
for (int j = 0; j < i; j++)
|
||||||
|
fprintf(stderr, "PASS: %s\n", nvccSteps[j].c_str());
|
||||||
|
fprintf(stderr, "FAIL: %s\n", nvccSteps[i].c_str());
|
||||||
|
exit(-1);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if (!keepTemporaries)
|
||||||
|
{
|
||||||
|
/* remove temporaries */
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
}
|
||||||
Reference in New Issue
Block a user