Module: Cumo::CUDA::CUDNN

Defined in:
lib/cumo/cuda/cudnn.rb,
ext/cumo/cuda/cudnn.c

Constant Summary collapse

CUDNN_POOLING_MAX = 
INT2NUM(CUDNN_POOLING_MAX)
CUDNN_POOLING_MAX_DETERMINISTIC = 
INT2NUM(CUDNN_POOLING_MAX_DETERMINISTIC)
CUDNN_POOLING_AVERAGE_COUNT_INCLUDE_PADDING = 
INT2NUM(CUDNN_POOLING_AVERAGE_COUNT_INCLUDE_PADDING)
CUDNN_POOLING_AVERAGE_COUNT_EXCLUDE_PADDING = 
INT2NUM(CUDNN_POOLING_AVERAGE_COUNT_EXCLUDE_PADDING)

Class Method Summary collapse

Class Method Details

.available? ⇒ Boolean

Returns availability of cuDNN.

Returns:

  • (Boolean) —

    Returns true if cuDNN is available



52
53
54
55
56
57
58
59
60
 
# File 'ext/cumo/cuda/cudnn.c', line 52

static VALUE
rb_cudnn_available_p()
{
#if CUDNN_FOUND
    return Qtrue;
#else
    return Qfalse;
#endif
}

.avg_pool(a, *args, **kwargs) ⇒ Object 



78
79
80
 
# File 'lib/cumo/cuda/cudnn.rb', line 78

def avg_pool(a, *args, **kwargs)
  a.avg_pool(*args, **kwargs)
end

.avg_pool_backward(a, *args, **kwargs) ⇒ Object 



82
83
84
 
# File 'lib/cumo/cuda/cudnn.rb', line 82

def avg_pool_backward(a, *args, **kwargs)
  a.avg_pool_backward(*args, **kwargs)
end

.batch_norm(a, *args, **kwargs) ⇒ Object 



62
63
64
 
# File 'lib/cumo/cuda/cudnn.rb', line 62

def batch_norm(a, *args, **kwargs)
  a.batch_norm(*args, **kwargs)
end

.batch_norm_backward(a, *args, **kwargs) ⇒ Object 



66
67
68
 
# File 'lib/cumo/cuda/cudnn.rb', line 66

def batch_norm_backward(a, *args, **kwargs)
  a.batch_norm_backward(*args, **kwargs)
end

.conv(a, *args, **kwargs) ⇒ Object 



48
49
50
 
# File 'lib/cumo/cuda/cudnn.rb', line 48

def conv(a, *args, **kwargs)
  a.conv(*args, **kwargs)
end

.conv_grad_w(a, *args, **kwargs) ⇒ Object Also known as: conv_backward_filter 



57
58
59
 
# File 'lib/cumo/cuda/cudnn.rb', line 57

def conv_grad_w(a, *args, **kwargs)
  a.conv_grad_w(*args, **kwargs)
end

.conv_transpose(a, *args, **kwargs) ⇒ Object Also known as: conv_backward_data 



52
53
54
 
# File 'lib/cumo/cuda/cudnn.rb', line 52

def conv_transpose(a, *args, **kwargs)
  a.conv_transpose(*args, **kwargs)
end

.max_pool(a, *args, **kwargs) ⇒ Object 



70
71
72
 
# File 'lib/cumo/cuda/cudnn.rb', line 70

def max_pool(a, *args, **kwargs)
  a.max_pool(*args, **kwargs)
end

.max_pool_backward(a, *args, **kwargs) ⇒ Object 



74
75
76
 
# File 'lib/cumo/cuda/cudnn.rb', line 74

def max_pool_backward(a, *args, **kwargs)
  a.max_pool_backward(*args, **kwargs)
end