タイムアウトでCUDA-Kernelを起動する

Question

私は特定のタイムアウトでCUDAカーネルを起動しようとしています。私はCUDAカーネルのデバイスタイムアウトがあることを知っていますが、共有サーバーで作業しているので、可能であってもこのタイムアウトを設定するアクセス権はありません。

私はこれをオートチューニングアプリケーションに必要とします。私は、すでに発見された最速のランタイムよりも速くならないカーネルの実行をキャンセルするタイムアウトを設定したいと思います。

このようなタイムアウトでCUDAカーネルを起動する方法はありますか？ ありがとうございます！

Answer 1

teraのリンクのおかげで、自分自身でタイムアウトを実装することができました。このスレッドで説明したように、次のように実行できます。

const int timeout = 2000000; 
int progressed = 0; 
while (cudaEventQuery(stop) != cudaSuccess) { 
    usleep(20000); 
    progressed += 20000; 
    if (progressed >= timeout) { 
     cudaDeviceReset(); 

     throw std::runtime_error("timeout"); 
    } 
} 
// No timeout occured 

この場合、stopはカーネル実行後に記録されたイベントです。