void CudaModuleScene::initCudaObj(ApexCudaSurfRef& surfRef)
{
if (getGpuDispatcher()->getCudaContextManager()->supportsArchSM20() == false)
{
return;
}
const char* surfRefName = surfRef.getName();
for (int j = 0 ; j < numRegisteredSurfaces ; j++)
{
if (nvidia::strcmp(surfaceTable[j].surfRefName, surfRefName) == 0)
{
ApexCudaModule* cudaModule = getCudaModule(surfaceTable[j].modIndex);
PX_ASSERT(cudaModule->isValid());
CUsurfref cuSurfRef;
CUT_SAFE_CALL(cuModuleGetSurfRef(&cuSurfRef, cudaModule->getCuModule(), surfRefName));
surfRef.init(this, cuSurfRef, cudaModule);
break;
}
}
}
void CudaModuleScene::initCudaObj(ApexCudaFunc& func)
{
for (int j = 0 ; j < numRegisteredFunctions ; j++)
{
const char* funcName = functionTable[j].funcName;
if (func.testNameMatch(funcName))
{
ApexCudaModule* cudaModule = getCudaModule(functionTable[j].modIndex);
PX_ASSERT(cudaModule->isValid());
CUfunction cuFunc = 0;
CUT_SAFE_CALL(cuModuleGetFunction(&cuFunc, cudaModule->getCuModule(), funcName));
PxU16 id = getGpuDispatcher()->registerKernelNames( &funcName, 1 );
func.init(this, funcName, cuFunc, cudaModule, id);
}
}
}
void CudaModuleScene::initCudaObj(ApexCudaVar& var)
{
const char* varName = var.getName();
for (int j = 0 ; j < numRegisteredVariables ; j++)
{
if (nvidia::strcmp(variableTable[j].varName, varName) == 0)
{
ApexCudaModule* cudaModule = getCudaModule(variableTable[j].modIndex);
PX_ASSERT(cudaModule->isValid());
CUdeviceptr cuDevPtr;
size_t size;
cuModuleGetGlobal(&cuDevPtr, &size, cudaModule->getCuModule(), varName);
var.init(this, cudaModule, cuDevPtr, size, getGpuDispatcher()->getCudaContextManager());
break;
}
}
}
