From dbf9c88aeddb4317fc56315d953eb3954997d0f2 Mon Sep 17 00:00:00 2001 From: tianchongchong <58495634+tianchongchong@users.noreply.github.com> Date: Tue, 25 Jun 2024 14:58:22 +0800 Subject: [PATCH] Update 05-device_target_interactions.md --- docs/arch/arch/05-device_target_interactions.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/docs/arch/arch/05-device_target_interactions.md b/docs/arch/arch/05-device_target_interactions.md index fc4c862f..7156113b 100644 --- a/docs/arch/arch/05-device_target_interactions.md +++ b/docs/arch/arch/05-device_target_interactions.md @@ -66,7 +66,7 @@ TVM 框架若要使用新的 DeviceAPI,应该按照以下步骤注册: ## Target 代码生成器 -代码生成器采用优化的 `IRModule`,并将其转换为可执行表示。每个代码生成器注册后,才能被 TVM 框架使用。这是通过注册 `"target.build.foo"` 函数来完成的,其中 `foo` 与上面的 `TVM_REGISTER_TARGET_KIND` 定义中使用的名称相同。 +代码生成器接收一个优化后的 `IRModule`,并将其转换为可执行表示。每个代码生成器注册后,才能被 TVM 框架使用。这是通过注册 `"target.build.foo"` 函数来完成的,其中 `foo` 与上面的 `TVM_REGISTER_TARGET_KIND` 定义中使用的名称相同。 ``` c++ tvm::runtime::Module GeneratorFooCode(IRModule mod, Target target);