为高性能计算系统打造的 AI 编码助手
HPC 编码助手是一款由 AI 赋能的强大工具,可直接在超级计算机的登录节点上运行,助力科学家、工程师和数据团队在基于 AMD 技术的 HPC 系统上高效开展工作。该工具不仅内置了通用编码智能体(例如 Claude Code 或 OpenCode),还深度融合了 HPC 专属知识、站点配套文档、作业调度方法以及 GPU 性能剖析指南。这样用户便能快速获得解答并顺畅执行任务,而无需不停地搜索信息。
所有功能均在容器内运行,无需在共享集群上安装。
主要功能特性
HPC 编码助手的核心功能
HPC 编码助手能够将自然语言指令转换为脚本、构建容器、管理资源、解释错误以及优化性能,从而帮助团队快速获得初始结果、轻松实现扩展、减少支持请求,同时高效利用计算资源。
咨询与系统相关的问题。
查询配额、分区、存储限制等信息,并基于站点提供的实际文档而非 AI 凭空推测做出响应。
编写和管理计算作业。
协助编写作业脚本、挑选合适的资源以及进行故障排查。
剖析 GPU 工作负载。
通过提供分步式指导,协助您逐步完成 AMD GPU 性能剖析。
加快完成常规编码任务。
查询配额、分区、存储限制等信息,并基于站点提供的实际文档而非 AI 凭空推测做出响应。
工作方式
构建容器。
使用已安装 Docker 的任何设备(例如您的笔记本电脑)构建容器,然后将容器传输到集群。
在登录节点上运行设置向导。
设置向导将引导您选择站点、AI 后端和首选智能体,然后自动处理其余操作。
通过单一脚本完成启动。
仅需一个脚本,即可处理日常启动操作,并在登录节点更改时保持服务正常运行。
当前版本限制
当前仅支持 LUMI
现有的技能和文档专为 LUMI 超级计算机量身定制。整体框架在设计层面能够支持其他站点,但尚未进行相应配置。
需要 AI 后端
需要 Aitta 令牌(针对 LUMI),或具有出站网络访问权限的 Anthropic API 密钥。
文档为快照
知识库在设置时构建,除非重新构建,否则不会接收上游更改。
周密考虑以实现安全保障
该工具能够根据指令避免执行破坏性操作,并在提交作业前请求确认。这种安全指令通过提示词下达,无法作为硬性安全防线。