Module: Cumo::CUDA::Runtime

Defined in:
ext/cumo/cuda/runtime.c

Class Method Summary collapse

Class Method Details

.cudaDeviceGetAttributes(attrib, device) ⇒ Object 



137
138
139
140
141
142
143
144
145
146
147
148
149
 
# File 'ext/cumo/cuda/runtime.c', line 137

static VALUE
rb_cudaDeviceGetAttributes(VALUE self, VALUE attrib, VALUE device)
{
    int _attrib = NUM2INT(attrib);
    int _device = NUM2INT(device);
    int _ret;
    cudaError_t status;

    status = cudaDeviceGetAttribute(&_ret, _attrib, _device);

    check_status(status);
    return INT2NUM(_ret);
}

.cudaDeviceSynchronize ⇒ Object 



189
190
191
192
193
194
195
196
 
# File 'ext/cumo/cuda/runtime.c', line 189

static VALUE
rb_cudaDeviceSynchronize(VALUE self)
{
    cudaError_t status;
    status = cudaDeviceSynchronize();
    check_status(status);
    return Qnil;
}

.cudaDriverGetVersion ⇒ Object 



81
82
83
84
85
86
87
88
89
90
91
 
# File 'ext/cumo/cuda/runtime.c', line 81

static VALUE
rb_cudaDriverGetVersion(VALUE self)
{
    int _version;
    cudaError_t status;

    status = cudaDriverGetVersion(&_version);

    check_status(status);
    return INT2NUM(_version);
}

.cudaGetDevice ⇒ Object 



122
123
124
125
126
 
# File 'ext/cumo/cuda/runtime.c', line 122

static VALUE
rb_cudaGetDevice(VALUE self)
{
    return INT2NUM(cumo_cuda_runtime_get_device());
}

.cudaGetDeviceCount ⇒ Object 



158
159
160
161
162
 
# File 'ext/cumo/cuda/runtime.c', line 158

static VALUE
rb_cudaGetDeviceCount(VALUE self)
{
    return INT2NUM(cumo_cuda_runtime_get_device_count());
}

.cudaRuntimeGetVersion ⇒ Object 



99
100
101
102
103
104
105
106
107
108
109
 
# File 'ext/cumo/cuda/runtime.c', line 99

static VALUE
rb_cudaRuntimeGetVersion(VALUE self)
{
    int _version;
    cudaError_t status;

    status = cudaRuntimeGetVersion(&_version);

    check_status(status);
    return INT2NUM(_version);
}

.cudaSetDevice(device) ⇒ Object 



171
172
173
174
175
176
177
178
179
180
181
 
# File 'ext/cumo/cuda/runtime.c', line 171

static VALUE
rb_cudaSetDevice(VALUE self, VALUE device)
{
    int _device = NUM2INT(device);
    cudaError_t status;

    status = cudaSetDevice(_device);

    check_status(status);
    return Qnil;
}