开发者
资源
[object Object][object Object][object Object]undefined
[object Object]
  • 接口功能:将张量x和配置参数(如param、cmd)发送至PMCC混淆引擎。引擎的CA模块调用TA模块,进行张量混淆处理,最终返回shape与x一致的混淆后的张量y。

  • 背景:PMCC(Privacy&Model Confidential Computing)模型混淆特性利用CPU核中的TrustZone可信执行环境隔离存储混淆因子、派生混淆掩码、执行动态掩码添加。PMCC基于NPU TrustZone构建了模型混淆引擎CA(普通OS中的Client Application)与模型混淆引擎TA(TEE OS中的Trusted Application)。为了使模型在推理执行过程中能够访问模型混淆引擎TA,通过AICPU算子机制及NPU卡内localhost socket进行中转。本次接口新增obfCoefficient,本参数为DeepSeek满血版本中,为了保证模型混淆性能满足要求,增加混淆系数,对输入数据按照混淆系数比例进行处理。

[object Object]

每个算子分为,必须先调用 “aclnnObfuscationCalculateV2GetWorkspaceSize”接口获取计算所需workspace大小以及包含了算子计算流程的执行器,再调用“aclnnObfuscationCalculateV2”接口执行计算。

[object Object]
[object Object]
[object Object]
  • 参数说明

    [object Object]
    • [object Object]Atlas 推理系列产品[object Object]:不支持BFLOAT16
  • 返回值

    aclnnStatus:返回状态码,具体参见

    第一段接口完成入参校验,出现以下场景时报错:

    [object Object]
[object Object]
  • 参数说明

    [object Object]
  • 返回值

    返回aclnnStatus状态码,具体参见

[object Object]
  • 确定性计算:

    • aclnnObfuscationCalculateV2默认确定性实现。
  • 该接口与配套使用,完成PMCC模型混淆功能,使用方式如下:

    • 首先调用aclnnObfuscationSetupV2进行资源初始化,可重复调用,以最后一次初始化为准
    • 再多次调用aclnnObfuscationCalculateV2进行张量混淆处理
    • 最后调用aclnnObfuscationSetupV2进行资源释放,只能调用一次;也可不显式进行资源释放,而是通过终止程序进程的方式达到资源释放的目的
[object Object]

调用示例代码如下,仅供参考,具体编译和执行过程请参考

[object Object]