Skip to content

当前为尝鲜版本,CANNBot-DSL 的 API 接口不保证兼容性,后续版本可能发生变更。

Host API ​

Host API 在普通 Python 代码或 Host 编排函数中使用,负责定义 kernel、准备编译参数、编译和加载产物、启动设备任务以及查询平台信息。

接口分类 ​

分类主要内容接口
装饰器与程序定义定义 Host 入口、DSL 辅助函数和设备 Kernelhost、jit、kernel 及其他装饰器
编译与运行编译、加载、调用和复用编译产物compile、load、ProviderCallable、clear_compile_cache
参数与数据描述描述编译期常量、动态维度和 Kernel 参数结构Constexpr、Dim、TensorSpec、TensorListSpec、StructSpec
平台与性能分析查询设备及存储能力并配置 profilingPlatformInfo、get_platform_info、get_mem_size、ProfileSpec、profiler.report_tensor_info
异常与诊断表达 Host 构图、编译和运行阶段错误CANNBotError、DiagnosticCode 等

CANNBot-DSL 开源项目文档