cuda - アクティブなCUDAデバイスからプロパティを取得するには?

Question

CUDA デバイスからプロパティを取得するためのよく知られたコード (!) は、すべてのデバイスを列挙し、そこからプロパティを取得します。次に、特定のデバイスをアクティブにするようなピースが表示されます。

逆に問題があります。デバイスが既に選択されていて、システムに存在するすべてのデバイスではなく、そのデバイス (アクティブなデバイス) のプロパティを取得したいとします。

私はCUDAが初めてなので、これを正しい方法で書いたことを願っています。

Answer 1

を呼び出しcudaGetDevice()てアクティブコンテキストのデバイス番号を取得してから、を呼び出しcudaGetDevicePropertiesてそのデバイスのプロパティを取得します。コードでは、次のようになります。

int device;
cudaGetDevice(&device);

struct cudaDeviceProp props;
cudaGetDeviceProperties(&props, device);

[免責事項：ブラウザで記述されており、コンパイルまたはテストされていません。自己責任で使用してください]