15 lines
301 B
Plaintext
15 lines
301 B
Plaintext
int get_device_attribute(
|
|
int attribute,
|
|
int device_id)
|
|
{
|
|
int device, value;
|
|
if (device_id < 0) {
|
|
cudaGetDevice(&device);
|
|
}
|
|
else {
|
|
device = device_id;
|
|
}
|
|
cudaDeviceGetAttribute(&value, static_cast<cudaDeviceAttr>(attribute), device);
|
|
return value;
|
|
}
|