nndeploy.device.device

Classes

Architecture(*args, **kwargs)
Device(*args, **kwargs)
DeviceInfo()
Event(device_or_type)
Stream(device_or_type[, stream])

class nndeploy.device.device.DeviceInfo

基类：DeviceInfo

__init__(self: nndeploy._nndeploy_internal.device.DeviceInfo) → None

property device_type

property is_support_fp16

property device_type_

property is_support_fp16_

class nndeploy.device.device.Architecture(*args, **kwargs)

基类：object

__init__(*args, **kwargs)

Constructs an Architecture object.

The constructor can be called in the following ways: 1. str: Constructs an Architecture from a string. eg: “cpu”, “ascendcl”, etc. 2. DeviceTypeCode: Constructs an Architecture from a DeviceTypeCode enum value. 3. _C.base.DeviceTypeCode: Constructs an Architecture from a DeviceTypeCode enum value.

check_device(device_id: int = 0, library_path: str = '')

enable_device(device_id: int = 0, library_path: str = '')

disable_device()

get_device(device_id: int)

get_device_info(library_path: str = '')

get_device_type_code()

insert_device(device_id: int, device: Device)

class nndeploy.device.device.Device(*args, **kwargs)

基类：object

__init__(*args, **kwargs)

Constructs a Device object.

The constructor can be called in the following ways: 1. Device(device_name_and_id): Constructs a Device from a string in the format “device_name:device_id” (e.g., “cuda:0”). 2. Device(device_name, device_id): Constructs a Device from a string device name and an integer device ID. 3. Device(device_type_code): Constructs a Device from a DeviceTypeCode enum value. The device ID defaults to 0. 4. Device(device_type_code, device_id): Constructs a Device from a DeviceTypeCode enum value and an integer device ID. 5. Device(): Constructs a default Device with DeviceTypeCode.cpu and device ID 0.

get_data_format_by_shape(shape)

to_buffer_desc(desc, config)

allocate(size_or_desc)

Allocate memory that must be freed using deallocate() :param size_or_desc: int for allocation size, or BufferDesc for buffer descriptor

返回:

Memory pointer that must be freed using deallocate()

deallocate(ptr)

allocate_pinned(size_or_desc)

Allocate pinned memory that must be freed using deallocate_pinned() :param size_or_desc: int for allocation size, or BufferDesc for buffer descriptor

返回:

Memory pointer that must be freed using deallocate_pinned()

deallocate_pinned(ptr)

copy(src, dst, size, stream=None)

download(src, dst, size, stream=None)

upload(src, dst, size, stream=None)

copy_buffer(src, dst, stream=None)

download_buffer(src, dst, stream=None)

upload_buffer(src, dst, stream=None)

get_context()

create_stream(stream=None)

create_event()

create_events(events)

get_device_type()

init()

deinit()

class nndeploy.device.device.Stream(device_or_type, stream=None)

基类：object

__init__(device_or_type, stream=None)

get_device_type()

get_device()

synchronize()

record_event(event)

wait_event(event)

on_execution_context_setup()

on_execution_context_teardown()

get_native_stream()

class nndeploy.device.device.Event(device_or_type)

基类：object

__init__(device_or_type)

get_device_type()

get_device()

query_done()

synchronize()

get_native_event()