Jaki jest kanoniczny sposób sprawdzania błędów za pomocą CUDA runtime API?

Powyższa odpowiedź Talonmies to świetny sposób na anulowanie aplikacji w stylu assert.

Czasami możemy chcieć zgłosić i odzyskać stan błędu w kontekście C++ jako część większej aplikacji.

Oto dość zwięzły sposób, aby to zrobić, rzucając wyjątek C++ pochodzący z std::runtime_error za pomocą thrust::system_error:

#include <thrust/system_error.h>
#include <thrust/system/cuda/error.h>
#include <sstream>

void throw_on_cuda_error(cudaError_t code, const char *file, int line)
{
  if(code != cudaSuccess)
  {
    std::stringstream ss;
    ss << file << "(" << line << ")";
    std::string file_and_line;
    ss >> file_and_line;
    throw thrust::system_error(code, thrust::cuda_category(), file_and_line);
  }
}

Spowoduje to włączenie nazwy pliku, numeru linii i opisu cudaError_t w języku angielskim do wyrzuconego wyjątku .what() członek:

#include <iostream>

int main()
{
  try
  {
    // do something crazy
    throw_on_cuda_error(cudaSetDevice(-1), __FILE__, __LINE__);
  }
  catch(thrust::system_error &e)
  {
    std::cerr << "CUDA error after cudaSetDevice: " << e.what() << std::endl;

    // oops, recover
    cudaSetDevice(0);
  }

  return 0;
}

Wyjście:

$ nvcc exception.cu -run
CUDA error after cudaSetDevice: exception.cu(23): invalid device ordinal

Klient some_function może odróżnić błędy CUDA od innych rodzajów błędów w razie potrzeby:

try
{
  // call some_function which may throw something
  some_function();
}
catch(thrust::system_error &e)
{
  std::cerr << "CUDA error during some_function: " << e.what() << std::endl;
}
catch(std::bad_alloc &e)
{
  std::cerr << "Bad memory allocation during some_function: " << e.what() << std::endl;
}
catch(std::runtime_error &e)
{
  std::cerr << "Runtime error during some_function: " << e.what() << std::endl;
}
catch(...)
{
  std::cerr << "Some other kind of error during some_function" << std::endl;

  // no idea what to do, so just rethrow the exception
  throw;
}

Ponieważ thrust::system_error jest std::runtime_error, możemy alternatywnie obsługiwać go w taki sam sposób jak szeroką klasę błędów, jeśli nie wymagamy precyzji poprzedniego przykładu:

try
{
  // call some_function which may throw something
  some_function();
}
catch(std::runtime_error &e)
{
  std::cerr << "Runtime error during some_function: " << e.what() << std::endl;
}

Sposób C++-canonical: nie sprawdzaj błędów...użyj wiązań C++, które rzucają wyjątki.

Kiedyś byłem irytowany tym problemem; i miałem rozwiązanie funkcji makro-cum-wrapper, tak jak w szponach i odpowiedziach Jareda,ale szczerze? To sprawia, że korzystanie z CUDA Runtime API jeszcze bardziej brzydkie i C-jak.

// (... prepare host-side buffers here ...)

auto current_device = cuda::device::current::get();
auto d_A = cuda::memory::device::make_unique<float[]>(current_device, numElements);
auto d_B = cuda::memory::device::make_unique<float[]>(current_device, numElements);
auto d_C = cuda::memory::device::make_unique<float[]>(current_device, numElements);

cuda::memory::copy(d_A.get(), h_A.get(), size);
cuda::memory::copy(d_B.get(), h_B.get(), size);

// (... prepare a launch configuration here... )
cuda::launch( vectorAdd, launch_config,
    d_A.get(), d_B.get(), d_C.get(), numElements
);    
cuda::memory::copy(h_C.get(), d_C.get(), size);

// (... verify results here...)

Ponownie - wszystkie potencjalne błędy są sprawdzane i zgłaszane za pomocą wyrzuconego wyjątku. Ten kod używa mojego

Thin Modern-wrappery C++ dla biblioteki CUDA Runtime API (Github)

Zauważ, że wyjątki zawierają zarówno wyjaśnienie ciągu znaków, jak i kod statusu CUDA runtime API po nieudanym wywołaniu.

Kilka linków do tego, jak błędy CUDA są automatycznie sprawdzane z tymi opakowaniami:

• test program rzuca i łapie kilka wyjątków
• dokumentacja funkcjonalności związanej z błędami

Rozwiązanie omówione Tutaj zadziałało dla mnie dobrze. To rozwiązanie wykorzystuje wbudowane funkcje cuda i jest bardzo proste do wdrożenia.

Odpowiedni kod jest skopiowany poniżej:

#include <stdio.h>
#include <stdlib.h>

__global__ void foo(int *ptr)
{
  *ptr = 7;
}

int main(void)
{
  foo<<<1,1>>>(0);

  // make the host block until the device is finished with foo
  cudaDeviceSynchronize();

  // check for error
  cudaError_t error = cudaGetLastError();
  if(error != cudaSuccess)
  {
    // print the CUDA error message and exit
    printf("CUDA error: %s\n", cudaGetErrorString(error));
    exit(-1);
  }

  return 0;
}