Skip to content

Commit 2eb35be

Browse files
committed
update
1 parent 8ebab06 commit 2eb35be

File tree

2 files changed

+8
-4
lines changed

2 files changed

+8
-4
lines changed

_posts/pytorch_read/2023-02-13-operators_register.md

+3-1
Original file line numberDiff line numberDiff line change
@@ -238,7 +238,9 @@ void OperatorEntry::registerSchema(FunctionSchema&& schema, std::string&& debug,
238238
}
239239
```
240240
registerSchema 首先遍历 kernels_ , 对 AnnotatedKernel 进行检查;
241-
然后调用 dispatchKeyExtractor_.registerSchema(schema)([参考 dispatcher](../2023-02-11-dispatcher#Dispatchkeyextractor)
241+
然后调用 dispatchKeyExtractor_.registerSchema(schema)
242+
<!-- ([参考 dispatcher](../2023-02-11-dispatcher#Dispatchkeyextractor) -->
243+
[参考 dispatcher]({% post_url pytorch_read/2023-02-11-dispatcher %})
242244
记录参数信息;
243245
最后生成成员变量 schema_ 。
244246

_posts/pytorch_read/2023-02-14-operators_call.md

+5-3
Original file line numberDiff line numberDiff line change
@@ -11,7 +11,7 @@ draft: false
1111

1212

1313
## 算子注册
14-
参考 [原生算子注册](../2023-02-13-operators_register)
14+
参考 [原生算子注册](../2023-02-13-operators_register.md)
1515
## 算子调用过程
1616
### 找到 OperatorHandle
1717
```cpp
@@ -84,6 +84,8 @@ C10_ALWAYS_INLINE_UNLESS_MOBILE Return Dispatcher::call(const TypedOperatorHandl
8484
return kernel.template call<Return, Args...>(op, dispatchKeySet, std::forward<Args>(args)...);
8585
}
8686
```
87-
上面的代码会使用 [dispatcher](../2023-02-11-dispatcher#Dispatchkeyextractor) 根据之前注册的信息生成一个 dispatchKeySet,然后根据之前注册的算子找到对应的函数运行。
87+
上面的代码会使用
88+
[dispatcher]({% post_url pytorch_read/2023-02-11-dispatcher %})
89+
根据之前注册的信息生成一个 dispatchKeySet,然后根据之前注册的算子找到对应的函数运行。
8890
### 自动梯度
89-
参考 [自动微分张量库](../2023-02-17-autograd)
91+
[自动微分张量库]({% post_url pytorch_read/2023-02-17-autograd %})

0 commit comments

Comments
 (0)