Пример #1
0
 public void SetAlgorithmDescriptor(cudnnAlgorithm algorithm)
 {
     res = CudaDNNNativeMethods.cudnnSetAlgorithmDescriptor(_desc, algorithm);
     Debug.WriteLine(String.Format("{0:G}, {1}: {2}", DateTime.Now, "cudnnSetAlgorithmDescriptor", res));
     if (res != cudnnStatus.Success)
     {
         throw new CudaDNNException(res);
     }
 }
Пример #2
0
        public cudnnAlgorithm GetAlgorithmDescriptor()
        {
            cudnnAlgorithm algo = new cudnnAlgorithm();

            res = CudaDNNNativeMethods.cudnnGetAlgorithmDescriptor(_desc, ref algo);
            Debug.WriteLine(String.Format("{0:G}, {1}: {2}", DateTime.Now, "cudnnSetAlgorithmDescriptor", res));
            if (res != cudnnStatus.Success)
            {
                throw new CudaDNNException(res);
            }
            return(algo);
        }