Ejemplos de cudaDeviceGetStreamPriorityRange en C++ (Cpp)

Ejemplo n.º 1

Archivo: common.cpp Proyecto: Caffe-MPI/Caffe-MPI.github.io

CudaStream::CudaStream(bool high_priority = false) {
  if (high_priority) {
    int leastPriority, greatestPriority;
    CUDA_CHECK(cudaDeviceGetStreamPriorityRange(&leastPriority, &greatestPriority));
    CUDA_CHECK(cudaStreamCreateWithPriority(&stream_, cudaStreamDefault, greatestPriority));
  } else {
    CUDA_CHECK(cudaStreamCreate(&stream_));
  }
  DLOG(INFO) << "New " << (high_priority ? "high priority " : "") << "stream "
      << stream_ << " on device " << current_device() << ", thread " << std::this_thread::get_id();
}

Ejemplo n.º 2

Archivo: task_scheduler.cpp Proyecto: dimaleks/uDeviceX

TaskScheduler::TaskScheduler()
{
    CUDA_Check( cudaDeviceGetStreamPriorityRange(&cudaPriorityLow, &cudaPriorityHigh) );
}