kernel_t<CUDA>* kernel_t<CUDA>::buildFromSource(const std::string &filename,
const std::string &functionName_,
const kernelInfo &info_){
OCCA_EXTRACT_DATA(CUDA, Kernel);
functionName = functionName_;
kernelInfo info = info_;
std::string cachedBinary = getCachedBinaryName(filename, info);
struct stat buffer;
const bool fileExists = (stat(cachedBinary.c_str(), &buffer) == 0);
if(fileExists){
std::cout << "Found cached binary of [" << filename << "] in [" << cachedBinary << "]\n";
return buildFromBinary(cachedBinary, functionName);
}
if(!haveFile(cachedBinary)){
waitForFile(cachedBinary);
return buildFromBinary(cachedBinary, functionName);
}
std::string iCachedBinary = createIntermediateSource(filename,
cachedBinary,
info);
std::string libPath, soname;
getFilePrefixAndName(cachedBinary, libPath, soname);
std::string oCachedBinary = libPath + "o_" + soname + ".o";
std::string archSM = "";
if(dev->dHandle->compilerFlags.find("-arch=sm_") == std::string::npos){
std::stringstream archSM_;
int major, minor;
OCCA_CUDA_CHECK("Kernel (" + functionName + ") : Getting CUDA Device Arch",
cuDeviceComputeCapability(&major, &minor, data_.device) );
archSM_ << " -arch=sm_" << major << minor << ' ';
archSM = archSM_.str();
}
std::stringstream command;
//---[ PTX Check Command ]----------
if(dev->dHandle->compilerEnvScript.size())
command << dev->dHandle->compilerEnvScript << " && ";
command << dev->dHandle->compiler
<< ' ' << dev->dHandle->compilerFlags
<< archSM
<< " -Xptxas -v,-dlcm=cg,-abi=no"
<< ' ' << info.flags
<< " -x cu -c " << iCachedBinary
<< " -o " << oCachedBinary;
const std::string &ptxCommand = command.str();
std::cout << "Compiling [" << functionName << "]\n" << ptxCommand << "\n";
#if (OCCA_OS == LINUX_OS) || (OCCA_OS == OSX_OS)
const int ptxError = system(ptxCommand.c_str());
#else
const int ptxError = system(("\"" + ptxCommand + "\"").c_str());
#endif
// Not needed here I guess
// if(ptxError){
// releaseFile(cachedBinary);
// throw 1;
// }
//---[ Compiling Command ]----------
command.str("");
command << dev->dHandle->compiler
<< " -o " << cachedBinary
<< " -ptx -I."
<< ' ' << dev->dHandle->compilerFlags
<< archSM
<< ' ' << info.flags
<< " -x cu " << iCachedBinary;
const std::string &sCommand = command.str();
std::cout << sCommand << '\n';
const int compileError = system(sCommand.c_str());
if(compileError){
releaseFile(cachedBinary);
throw 1;
}
const CUresult moduleLoadError = cuModuleLoad(&data_.module,
cachedBinary.c_str());
if(moduleLoadError)
releaseFile(cachedBinary);
OCCA_CUDA_CHECK("Kernel (" + functionName + ") : Loading Module",
moduleLoadError);
const CUresult moduleGetFunctionError = cuModuleGetFunction(&data_.function,
data_.module,
functionName.c_str());
if(moduleGetFunctionError)
releaseFile(cachedBinary);
OCCA_CUDA_CHECK("Kernel (" + functionName + ") : Loading Function",
moduleGetFunctionError);
releaseFile(cachedBinary);
return this;
}
kernel_t<COI>* kernel_t<COI>::buildFromSource(const std::string &filename,
const std::string &functionName_,
const kernelInfo &info_){
functionName = functionName_;
kernelInfo info = info_;
info.addDefine("OCCA_USING_CPU", 1);
info.addDefine("OCCA_USING_COI", 1);
info.addOCCAKeywords(occaCOIDefines);
std::stringstream salt;
salt << "COI"
<< info.salt()
<< dev->dHandle->compilerEnvScript
<< dev->dHandle->compiler
<< dev->dHandle->compilerFlags
<< functionName;
std::string cachedBinary = getCachedName(filename, salt.str());
std::string libPath, soname;
getFilePrefixAndName(cachedBinary, libPath, soname);
std::string libName = "lib" + soname + ".so";
cachedBinary = libPath + libName;
struct stat buffer;
bool fileExists = (stat(cachedBinary.c_str(), &buffer) == 0);
if(fileExists){
std::cout << "Found cached binary of [" << filename << "] in [" << cachedBinary << "]\n";
return buildFromBinary(cachedBinary, functionName);
}
if(!haveFile(cachedBinary)){
waitForFile(cachedBinary);
return buildFromBinary(cachedBinary, functionName);
}
std::string iCachedBinary = createIntermediateSource(filename,
cachedBinary,
info);
std::stringstream command;
if(dev->dHandle->compilerEnvScript.size())
command << dev->dHandle->compilerEnvScript << " && ";
command << dev->dHandle->compiler
#if (OCCA_OS == LINUX_OS) || (OCCA_OS == OSX_OS)
<< " -x c++ -w -nodefaultlibs -fPIC -shared"
#else
<< " /TP /LD /D MC_CL_EXE"
#endif
<< ' ' << dev->dHandle->compilerFlags
<< ' ' << info.flags
<< ' ' << iCachedBinary
#if (OCCA_OS == LINUX_OS) || (OCCA_OS == OSX_OS)
<< " -o " << cachedBinary
#else
<< " /link /OUT:" << cachedBinary
#endif
<< std::endl;
const std::string &sCommand = command.str();
std::cout << "Compiling [" << functionName << "]\n" << sCommand << "\n";
const int compileError = system(sCommand.c_str());
if(compileError){
releaseFile(cachedBinary);
throw 1;
}
OCCA_EXTRACT_DATA(COI, Kernel);
COILIBRARY outLibrary;
const COIRESULT loadingLibraryResult = COIProcessLoadLibraryFromFile(data_.chiefID,
cachedBinary.c_str(),
soname.c_str(),
NULL,
&outLibrary);
if(errorCode != COI_SUCCESS)
releaseFile(cachedBinary);
OCCA_COI_CHECK("Kernel: Loading Kernel To Chief", loadingLibraryResult);
const char *c_functionName = functionName.c_str();
const COIRESULT getFunctionHandleResult = COIProcessGetFunctionHandles(data_.chiefID,
1,
&c_functionName,
&(data_.kernel));
if(errorCode != COI_SUCCESS)
releaseFile(cachedBinary);
OCCA_COI_CHECK("Kernel: Getting Handle", getFunctionHandleResult);
releaseFile(cachedBinary);
return this;
}
