• M
    fix(autodiff): fix some bugs in relation to 2nd order grad · 273c0e87
    Megvii Engine Team 提交于
    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
dispatch.cpp 3.5 KB