提交 · 273c0e874548a361149e73166b70cb963af5c878 · MegEngine 天元 / MegEngine

23 3月, 2022 1 次提交

fix(autodiff): fix some bugs in relation to 2nd order grad · 273c0e87

由 Megvii Engine Team 提交于 3月 01, 2022

1. implement double backward for batchnorm
2. fix grad attach in nested grad manager
3. pad empty tensor for unsatisfied output_has_grad
4. support double backward for jit subgraph
5. support double backward for autodiff.Function
6. readd debug flag MGE_LOG_OP_DISPATCH

GitOrigin-RevId: cd31ddc620a35e0582c9721df7290c972fa3c610

273c0e87

21 3月, 2022 3 次提交
- M
  perf(imperative): add dim_expansion transform for conv/bn1d · a4327c4d
  由 Megvii Engine Team 提交于 3月 17, 2022
```
GitOrigin-RevId: d14a69424d9e15ac9ae29a9e4bcd9b532dc76200
```
  a4327c4d
- M
  fix(imperative): fix the dtype promote problem when amp · e64536a3
  由 Megvii Engine Team 提交于 3月 11, 2022
```
GitOrigin-RevId: 43e1035fc86bc5d9212f5de00e4cce347940a7dd
```
  e64536a3
- M
  refactor(imperative): move python code of elemwise/reduce/conv2d/bn to c++ · 3c3fc6f3
  由 Megvii Engine Team 提交于 2月 23, 2022
```
GitOrigin-RevId: 01b532439243aa2e7d40f150fcaa26fded0e4f27
```
  3c3fc6f3
25 2月, 2022 3 次提交
- M
  refactor(dispatch): allow dynamic type creation · 177001d5
  由 Megvii Engine Team 提交于 2月 16, 2022
```
GitOrigin-RevId: 27dde05cff7e1e0bf61c652a3ae1fe4def829ada
```
  177001d5
- M
  perf(dispatch): improve performance of dispatch system · 4fa61620
  由 Megvii Engine Team 提交于 1月 27, 2022
```
GitOrigin-RevId: 860028e1af63936e7b4edefbed90d8244e7cb8d2
```
  4fa61620
- M
  perf(dispatch): speed up dispatch system · ca001777
  由 Megvii Engine Team 提交于 1月 20, 2022
```
GitOrigin-RevId: eabbe3e0219ff989801751c726eb0828b1b7a740
```
  ca001777
15 1月, 2022 1 次提交
- M
  feat(imperative/python): add transformation manager · d3689c3f
  由 Megvii Engine Team 提交于 1月 14, 2022
```
GitOrigin-RevId: a3c1732ffd4fafdde27c69fcdc3488ae19ec3dd0
```
  d3689c3f

MegEngine 天元 / MegEngine 1 年多 前同步成功

MegEngine 天元 / MegEngine
1 年多前同步成功