Searched refs:deviceNum (Results 1 – 1 of 1) sorted by relevance
380 PetscErrorCode setupStream(PetscInt deviceNum, PetscBool runDouble, PetscBool cpuTiming) in setupStream() argument395 if (deviceNum >= deviceCount || deviceNum < 0) { in setupStream()396 …LF, "\n!!!!!Invalid GPU number %d given hence default gpu %d will be used !!!!!\n", deviceNum, 0)); in setupStream()397 deviceNum = 0; in setupStream()401 cudaSetDevice(deviceNum); in setupStream()404 if (cudaGetDeviceProperties(&deviceProp, deviceNum) != cudaSuccess) { in setupStream()