选取昇腾AI处理器

任务说明

Volcano框架根据节点优选得到分数后为Pod任务选择最优的节点,并将其绑定。在该过程中可以注册回调函数,依据亲和性策略,实现昇腾AI处理器选取。

例如Pod任务需要1个昇腾AI处理器,此时节点两个HCCL环分别剩余1个和3个昇腾AI处理器,那么最终会选择剩余1个昇腾AI处理器的环。

具体实现

具体代码实现请参考开源代码中UseAnnotation方法,其中SelectNPUFromNode方法实现了根据亲和性从node上选取昇腾AI处理器的功能。