backend.ai-kernel-runner 1.4.2

如何编写新的计算内核

继承ai.backend.kernel.BaseRunner并实现以下方法

• async def init_with_loop(self)

  • 在asyncio事件循环可用后调用。

  • 大部分只是pass。

  • 如果您的内核支持交互式用户输入，那么将self.user_input_queue作为一个asyncio.Queue对象。这是您的工作，利用队列对象等待用户输入。（参见ai/backend/kernel/python/inproc.py中的handle_input()方法作为参考）如果没有设置，那么尝试获取交互式用户输入将直接返回"<user-input is unsupported>"。

• async def build_heuristic(self)

  • （批量模式）编写一个启发式代码，找到一些构建脚本或运行足够好的构建命令。

  • （阻塞）您不必担心重叠执行，因为基础运行器会处理。

• async def execute_heuristic(self)

  • （批量模式）编写一个启发式代码，找到主程序。

  • （阻塞）您不必担心重叠执行，因为基础运行器会处理。

• async def query(self, code_text)

  • （查询模式）直接运行给定的代码片段。根据语言/运行时，您可能需要创建一个临时文件并执行外部程序。

  • （阻塞）您不必担心重叠执行，因为基础运行器会处理。

• async def complete(self, data)

  • (查询模式) 接收一个包含用户当前输入代码行的字典数据，并返回一个可以自动完成的字符串列表。

  • (非阻塞) 应该实现此方法以异步方式与代码执行同时运行。

• async def interrupt(self)

  • (查询模式) 向正在运行的程序发送中断信号。实现方式由您决定。当前的Python运行器为进程内的查询模式执行启动一个线程，并使用ctypes黑客技术将其中的KeyboardInterrupt异常抛出。

  • (非阻塞) 应该实现此方法以异步方式与代码执行同时运行。

注意：现有代码是良好的参考！

如何在Backend.AI计算内核中使用它

使用pip通过Dockerfile中的RUN指令安装此软件包。然后，设置如下的CMD指令

CMD ["/home/backend.ai/jail", "-policy", "/home/backend.ai/policy.yml", \
     "/usr/local/bin/python", "-m", "ai.backend.kernel", "<language>"]

其中<language>应该是ai/backend/kernel/__main__.py文件中定义的lang_map变量支持的编程语言名称之一。