Class KirinNPUDeviceInfo

Inheritance Relationships

Base Type

Class Documentation

class KirinNPUDeviceInfo : public mindspore::DeviceInfoContext

Derived from DeviceInfoContext, The configuration of the model running on the NPU. This option is only valid for MindSpore Lite.

Public Functions

inline virtual enum DeviceType GetDeviceType() const override

Get the type of this DeviceInfoContext.

Returns

Type of this DeviceInfoContext.

void SetEnableFP16(bool is_fp16)

Set enables to perform the float16 inference.

Parameters

is_fp16 – [in] Enable float16 inference or not.

bool GetEnableFP16() const

Get enables to perform the float16 inference.

Returns

Whether enable float16 inference.

void SetFrequency(int frequency)

Set the NPU frequency.

Parameters

frequency – [in] Can be set to 1 (low power consumption), 2 (balanced), 3 (high performance), 4 (extreme performance), default as 3.

int GetFrequency() const

Get the NPU frequency.

Returns

NPU frequency