[发明专利]基于x86架构的AI芯片任务处理方法和装置

权利要求书

1.一种基于x86架构的AI芯片任务处理方法，其特征在于，所述方法包括：

更改内核配置文件，并基于更改的内核配置文件对x86系统进行内核的重新编译；

利用预设编译器版本和库文件对x86系统进行编译环境搭建；

在已搭建的编译环境中对AI芯片进行设备注册以生成AI芯片驱动，并将所述AI芯片驱动挂载到x86系统；

获取计算任务，并利用x86系统的运行时程序将所述计算任务下传至所述AI芯片驱动；

利用所述AI芯片驱动和所述计算任务对AI芯片的功能模块进行任务配置和使能直至任务完成。

2.根据权利要求1所述的方法，其特征在于，所述更改内核配置文件，并基于更改的内核配置文件对x86系统进行内核的重新编译的步骤包括：

获取底层配置文件，将所述底层配置文件中的drm功能和dma功能均配置为打开；

获取顶层配置文件，将所述顶层配置文件中的dma连续内存分配功能配置为打开；

利用所述底层配置文件和所述顶层配置文件对x86系统进行内核的重新编译。

3.根据权利要求1所述的方法，其特征在于，所述利用预设编译器版本和库文件对x86系统进行编译环境搭建的步骤包括：

获取x86系统的当前编译器版本，并将所述当前编译器版本与所述预设编译器版本进行比较；

若所述当前编译器版本低于所述预设编译器版本，则将当前编译器版本替换为所述预设编译器版本；

获取jpeg包和png包，并将所述jpeg包和png包添加至x86系统。

4.根据权利要求1所述的方法，其特征在于，对AI芯片进行设备注册包括：PCI驱动定义、AI芯片ID声明、AI芯片配置、任务队列初始化、AI芯片地址映射、AI芯片中断注册、dma字符设备添加、AI芯片dma连续内存分配。

5.根据权利要求4所述的方法，其特征在于，所述在已搭建的编译环境中对AI芯片进行设备注册以生成AI芯片驱动具体包括：

使用pci_driver驱动结构体将AI硬件注册为PCI设备，所述pci_driver驱动结构体包括AI芯片的名称、待驱动AI芯片列表、初始化AI芯片以及卸载AI芯片功能函数；

配置AI芯片厂商ID和AI芯片ID以进行PCI设备匹配；

为PCI设备定义私有的设备信息结构体，所述设备信息结构体用于接收保存PCI设备的中断信息、硬件配置信息、队列信息以及中断自旋锁信息，利用devm_kzalloc命令为所述设备信息结构体申请内存，利用pci_set_drvdata命令将所述设备信息结构体挂载到匹配的PCI设备上；

配置AI芯片的MAC数量、是否支持压缩、winograd算法进行初始化；

初始化AI芯片的任务队列；

利用APIpci_enable_device命令进行对应PCI设备的匹配与唤醒，利用pci_resource_start命令获取PCI设备的地址空间，利用ioremap命令将该地址进行虚拟地址映射，利用reg_read命令以及reg_write命令对相应的寄存器进行读写实现CPU与AI芯片之间的交互；

获取PCI设备的irq变量值，利用devm_request_irq命令来实现的对所述irq变量值和中断函数之间的对应匹配；

为AI芯片添加drm设备功能，其中drm设备注册了包含特定名称的字符设备，并为所述字符设备添加打开、文件操作以及ioctl功能；

为AI芯片分配dma存储空间，所述dma存储空间用于存放推理权值、输入、输出数据。

6.根据权利要求1所述的方法，其特征在于，所述获取计算任务，并利用x86系统的运行时程序将所述计算任务下传至所述AI芯片驱动的步骤包括：

利用x86系统的解析器对待处理任务进行解析以得到计算任务，并将所述计算任务发送至x86系统的运行时程序；

x86系统的运行时程序与所述AI芯片驱动进行交互以将所述计算任务发送至所述AI芯片驱动。