set_current_compile_soc_info
Description
Sets Ascend AI Processor version.
Prototype
def set_current_compile_soc_info(soc_version, core_type="AiCore", aicore_num=None, l1_fusion=None)
Parameters
Parameter |
Type |
Description |
|---|---|---|
soc_version |
(Required) String |
Indicates the Ascend AI Processor version. |
core_type |
(Optional) String |
Indicates the core type, selected from AiCore and VectorCore. Defaults to AiCore. |
aicore_num |
(Optional) Integer |
Indicates the number of AI Cores to be used. The value ranges from 1 to the available number of AI Cores. Defaults to the number of available AI Cores. |
l1_fusion |
(Optional) Bool |
Specifies whether to perform L1 fusion, either True or False. Defaults to False. |
Returns
success on success; error on failure.
Restrictions
None
Example
Before calling, replace soc_version with the actual Ascend AI Processor version.
import tbe soc_version="xxx" tbe.common.platform.set_current_compile_soc_info(soc_version)
Parent topic: TBE Platform API