[发明专利]使用内核加密数据的方法和系统

说明书

在一个实施方式中，数据处理(DP)加速器加密或解密输入数据的计算机实现的方法可以包括从主机装置接收命令、输入数据和内核。内核可以是加密内核或解密内核，并且DP加速器不需要知道其已经接收了哪个内核。DP加速器运行所接收的内核。响应于DP加速器接收到命令，如果接收到的内核是加密内核，DP加速器使用内核执行输入数据的加密，否则使用内核解密输入数据。然后将加密或解密的输入数据提供给主机装置。

技术领域

本公开的实施方式总体上涉及人工智能模型训练和推断。更具体地，本公开的实施方式涉及由数据处理加速器执行的人工智能模型训练和推理以及相关联的安全性。

背景技术

因为人工智能(AI)技术已经被部署在诸如图像分类或自动驾驶的各种领域中，AI模型(也称为“机器学习模型”)最近已经被广泛地使用。与软件应用的可执行图像或二进制图像类似，AI模型在被训练时可以基于一组属性执行推断以分类为特征。因此，AI模型可以是“便携式的”并且可以在没有授权的情况下使用。当前，缺乏对AI模型的有效的数字权利保护。此外，使用委托给诸如数据处理(DP)加速器或远程系统的二级处理系统的AI模型的处理任务，缺乏证据证明由DP加速器系统产生的结果受“信任根”系统保护。

发明内容

根据本申请的一个方面，公开了一种数据处理加速器加密或解密输入数据的计算机实现的方法，所述方法可包括：从主机装置接收命令、所述输入数据和内核；响应于所述数据处理加速器接收到所述命令：检查所述内核以确定所述内核的类型，以及响应于确定所接收的内核是加密内核由所述数据处理加速器使用所述内核加密所述输入数据，响应于确定所接收的内核是解密内核使用所述内核解密所述输入数据；以及将加密或解密的输入数据传输到所述主机装置。

根据本申请的另一方面，公开了一种数据处理加速器，其可包括：接口，从主机装置接收命令、输入数据和内核；以及安全单元，配置为响应于所述数据处理加速器接收到所述命令：检查所述内核以确定所述内核的类型，以及响应于确定所接收的内核是加密内核，使用所述内核加密所述输入数据，响应于确定所接收的内核是解密内核，使用所述内核解密所述输入数据，以及将加密或解密的输入数据发送到所述主机装置。

根据本申请的又一方面，公开了一种其中存储有指令的非暂时性的机器可读介质，所述指令在由处理器执行时使所述处理器执行数据处理加速器的操作。所述操作可包括：从主机装置接收命令、输入数据和内核；响应于所述数据处理加速器接收到命令：检查所述内核以确定所述内核的类型，以及响应于确定所接收的内核是加密内核，由所述数据处理加速器使用所述内核加密所述输入数据，响应于确定所接收的内核是解密内核，使用所述内核解密所述输入数据；以及将加密或解密的输入数据传输到所述主机装置。

根据本申请的再一方面，公开了一种主机装置，其可包括：处理器；以及存储器，联接到所述处理器以存储指令，所述指令在由所述处理器执行时致使所述处理器执行操作。所述操作可包括：由所述主机装置向数据处理加速器传输命令、输入数据和内核，其中所述内核是加密内核或解密内核，以及由所述主机装置从所述数据处理加速器获取加密或解密的输入数据，其中所述数据处理加速器被配置为响应于所述命令使用所述内核加密或解密所述输入数据。

附图说明

本公开的实施方式在附图的各图中以举例而非限制的方式示出，附图中的相同的附图标记指代相似的元件。

图1是示出根据一个实施方式的安全处理系统的框图。

图2A和图2B是示出根据一个实施方式的在一个或多个主机和一个或多个数据处理加速器之间的安全计算环境的框图。

图3A和图3B是示出根据实施方式的确保数据处理(DP)加速器对人工智能(AI)模型的水印进行签名的方法的框图。

图4A和图4B是示出根据实施方式的使用具有密码模块的DP加速器安全地加密或解密数据的方法的框图。