接口功能:将张量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进行中转。
每个算子分为,必须先调用 “aclnnObfuscationCalculateGetWorkspaceSize”接口获取计算所需workspace大小以及包含了算子计算流程的执行器,再调用“aclnnObfuscationCalculate”接口执行计算。
[object Object]
[object Object]
参数说明
[object Object]- [object Object]Atlas 推理系列产品[object Object]:不支持BFLOAT16
返回值
第一段接口完成入参校验,出现以下场景时报错:
[object Object]
确定性计算:
- aclnnObfuscationCalculate默认确定性实现。
-
- 首先调用aclnnObfuscationSetup进行资源初始化,可重复调用,以最后一次初始化为准
- 再多次调用aclnnObfuscationCalculate进行张量混淆处理
- 最后调用aclnnObfuscationSetup进行资源释放,只能调用一次;也可不显式进行资源释放,而是通过终止程序进程的方式达到资源释放的目的
[object Object]