Browse Source

Revert "fix(mgb/opr): change EQ opr's backward_graph to nullptr instead of InvalidGrad"

This reverts commit 6355337641.

GitOrigin-RevId: 6897f6d45d
tags/v1.0.0-rc1
Megvii Engine Team Xinran Xu 4 years ago
parent
commit
30ce3c60bd
2 changed files with 1 additions and 4 deletions
  1. +0
    -3
      src/opr/impl/dnn/roi_align.cpp
  2. +1
    -1
      src/opr/impl/dnn/roi_pooling.cpp

+ 0
- 3
src/opr/impl/dnn/roi_align.cpp View File

@@ -42,9 +42,6 @@ SymbolVar ROIAlignForward::make(SymbolVar src, SymbolVar rois,

#ifdef MGB_ENABLE_GRAD
MGB_IMPL_OPR_GRAD(ROIAlignForward) {
if (out_grad[1]) {
return InvalidGrad::make(opr, wrt_idx);
}
if (wrt_idx == 0) {
// wrt src
SymbolVar grad =


+ 1
- 1
src/opr/impl/dnn/roi_pooling.cpp View File

@@ -86,7 +86,7 @@ size_t ROIPoolingForward::get_workspace_size_bytes(

#ifdef MGB_ENABLE_GRAD
MGB_IMPL_OPR_GRAD(ROIPoolingForward) {
if (out_grad[1] || wrt_idx == 2) {
if (wrt_idx == 2) {
return InvalidGrad::make(opr, wrt_idx);
}
if (wrt_idx == 0) {


Loading…
Cancel
Save