예제 #1
0
 public GpuContext(Gpu gpu, bool useDefaultStream = false)
 {
     Device = Device.GpuDevice(gpu);
     Gpu    = gpu;
     if (useDefaultStream)
     {
         throw new NotImplementedException("TODO: add default stream support.");
     }
     else
     {
         Stream = gpu.CreateStream();
         if (Alea.cuDNN.Dnn.IsAvailable)
         {
             Dnn = new Alea.cuDNN.Dnn(Stream);
         }
         if (Alea.cuBLAS.Blas.IsAvailable)
         {
             Blas = new Alea.cuBLAS.Blas(Stream);
         }
     }
 }