mindspore.hal.get_device_properties

mindspore.hal.get_device_properties(device_id, device_target=None)[source]

Get specified device's properties, this api will be deprecated and removed in future versions.

Note

For Ascend, device must be initialized before calling this method, or total_memory and free_memory will be 0, and device_id will be ignored since this method only returns current device's properties.

Parameters

device_id (int) – The device id of which the properties will be returned.
device_target (str, optional) – The target device specified, should be one of "CPU" , "GPU" and "Ascend" . Default None , represents the current device set by context.

Returns

cudaDeviceProp for GPU.

cudaDeviceProp {
    name(str),
    major(int),
    minor(int),
    is_multi_gpu_board(int),
    is_integrated(int),
    multi_processor_count(int),
    total_memory(int),
    warp_size(int)
}

AscendDeviceProperties for Ascend.

AscendDeviceProperties {
    name(str),
    total_memory(int),
    free_memory(int)
}

None for CPU.

Examples

>>> import mindspore
>>> device_target = mindspore.context.get_context("device_target")
>>> print(mindspore.hal.get_device_properties(0, device_target))