private List <int> getGpus(int nMax)
{
CudaDnn <float> cuda = new CudaDnn <float>(0);
List <int> rgGpu = new List <int>();
int nDevCount = cuda.GetDeviceCount();
for (int i = 0; i < nDevCount; i++)
{
string strDevInfo = cuda.GetDeviceInfo(i, true);
string strP2PInfo = cuda.GetDeviceP2PInfo(i);
if (strP2PInfo.Contains("P2P Capable = YES"))
{
rgGpu.Add(i);
}
if (rgGpu.Count == nMax)
{
break;
}
}
cuda.Dispose();
return(rgGpu);
}
static void Main(string[] args)
{
// This memory will reside on the GPU.
long hGpuMem = 0;
Console.WriteLine("Creating CudaCuDnn...");
CudaDnn <float> cuda = new CudaDnn <float>(0);
try
{
string strDeviceInfo = cuda.GetDeviceName(0);
Console.WriteLine(strDeviceInfo);
strDeviceInfo = cuda.GetDeviceP2PInfo(0);
Console.WriteLine(strDeviceInfo);
strDeviceInfo = cuda.GetDeviceInfo(0, true);
Console.WriteLine(strDeviceInfo);
List <long> rghGpuMem = new List <long>();
long lOffset = 0;
// You must first allocate the GPU memory to use.
// Below we will allocate an array of 1000 float values.
Console.WriteLine("Allocate 1000 items...");
hGpuMem = cuda.AllocMemory(1000);
cuda.set(1000, hGpuMem, 0.0);
Console.WriteLine("Create memory pointers...");
for (int i = 0; i < 10; i++)
{
long hMem1 = cuda.CreateMemoryPointer(hGpuMem, lOffset, 100);
cuda.set(100, hMem1, (double)(i + 1));
rghGpuMem.Add(hMem1);
lOffset += 100;
}
Console.WriteLine("Test memory...");
for (int i = 0; i < 10; i++)
{
long hMem1 = rghGpuMem[i];
float[] rgData = cuda.GetMemoryFloat(hMem1);
if (rgData.Length != 100)
{
throw new Exception("The data length should = 100!");
}
for (int j = 0; j < 100; j++)
{
if (rgData[j] != (float)(i + 1))
{
throw new Exception("The data at index " + j.ToString() + " is not correct!");
}
}
}
Console.WriteLine("Memory test passed successfully!");
}
catch (Exception excpt)
{
Console.WriteLine("ERROR: " + excpt.Message);
}
finally
{
// Clean-up and release all GPU memory used.
if (hGpuMem != 0)
{
cuda.FreeMemory(hGpuMem);
hGpuMem = 0;
}
cuda.Dispose();
}
Console.WriteLine("Press any key to exit.");
Console.Read();
}