Forge CLI

Forge CLI

您GPU的新挚友:一款AI驱动的编辑器,让CUDA内核如歌似舞。

Forge CLI是一款专为NVIDIA GPU内核开发设计的AI驱动代码编辑器与优化工具。它集成了智能性能分析、自动化基准测试和硬件感知的AI建议,能显著加速CUDA、Triton和PyTorch代码的编写、测试与优化过程,以实现极致性能。

Freemium
Forge CLI screen shot

如何使用 Forge CLI?

安装Forge CLI并将其集成到您的开发工作流中。在编辑器中编写GPU内核代码(CUDA、Triton、PyTorch)。AI会在您键入时提供内联性能指标与优化建议。使用性能分析终端分析瓶颈,自动在不同GPU配置间进行基准测试,并模拟您未实际拥有的硬件以测试兼容性与性能。

Forge CLI 的核心功能

  • 内联性能分析与指标:在代码编辑器中直接查看内核执行时间、内存使用率和占用率等性能指标,无需单独运行性能分析。
  • 硬件感知AI:AI理解特定GPU架构(如A100、H100等),并提供针对您具体硬件的优化建议,包括内存布局、块大小和精度设置。
  • GPU模拟器:在50多种不同GPU架构(如H100和A100)上测试您的内核,误差低于2%,无需拥有实体硬件,实现广泛的兼容性测试。
  • 本地LLM支持:通过Ollama、vLLM或LM Studio在本地运行AI模型,确保您的专有代码完全私密,永不离开您的机器,保障数据安全。
  • 自动化基准测试与参数扫描:自动扫描不同配置(块大小、线程数、内存布局)以找到最快设置,并跟踪长期性能回归。
  • 自然语言性能分析:用通俗英语描述您想分析的内容,Forge CLI即刻生成正确且复杂的Nsight Compute命令,无需记忆繁琐参数。
  • 多GPU对比与分析:同时对比多个GPU上的内核性能(专业版最多6个,企业版无限制),为您的负载确定最佳硬件并优化扩展性。
  • Forge CLI 的使用场景

  • GPU内核开发者:通过即时反馈和AI驱动建议,加速高性能CUDA和Triton内核的开发与优化,将调试时间从数小时缩短至几分钟。
  • 机器学习工程师与研究人员:针对特定GPU硬件优化PyTorch模型训练和推理循环,自动建议混合精度、内核融合和内存管理改进,以加速实验进程。
  • 硬件验证工程师:在硬件到位或采购前,使用GPU模拟器在广泛的NVIDIA架构上测试软件兼容性与性能,降低部署风险。
  • 高性能计算应用开发者:在多GPU设置下分析和基准测试复杂的科学计算应用,识别瓶颈并高效优化以适配数据中心级部署。
  • AI基础设施团队:确保自定义AI模型和框架为其特定的数据中心GPU集群(如H100集群)进行最优调优,最大化昂贵硬件的利用率和投资回报率。
  • 学生与教育工作者:提供一个易于上手的集成环境,通过实时反馈和解释来学习GPU编程,降低并行计算概念的入门门槛。
  • Forge CLI 的价格

    FREE

    $0/mo

    适用于独立开发者。包含单GPU开发、无限性能分析与基准测试、CodeLens性能指标、GPU虚拟化、本地LLM支持,以及每月1个Forge积分。

    PRO

    $29/mo

    适用于专业团队。包含免费版所有功能,外加GPU模拟器访问(50多种GPU型号)、多GPU对比(最多6个)、自然语言性能分析、每月10个Forge积分、无限自动补全、GPU优化建议以及优先电子邮件支持。

    ENTERPRISE

    Custom Pricing

    适用于大型组织。包含专业版所有功能,外加100+ GPU集群支持、数据中心优化、本地化部署、定制芯片支持、无限Forge积分、自定义模型微调,以及专属的24/7 SLA技术支持。

    PAY AS YOU GO

    from $112.50

    用于AI驱动内核优化的智能体积分。若性能未超越torch.compile(mode='max-autotune'),积分将退还。包含数据中心级GPU(B200、H100、H200)的访问权限。

    Forge CLI 的常见问题

    最受影响的职业

    GPU Software Engineer
    Machine Learning Engineer
    High-Performance Computing (HPC) Developer
    AI Research Scientist
    Hardware Validation Engineer
    DevOps Engineer (AI/ML Infrastructure)
    Quantitative Developer
    Computer Vision Engineer
    Autonomous Systems Developer
    Computational Physics Researcher

    Forge CLI 的标签

    Forge CLI 的替代品