pytorch

ab7f5fd2 - [functorch] add batching rule for block_diag, kill DECOMPOSE_FUNCTIONAL (pytorch/functorch#814)

[functorch] add batching rule for block_diag, kill DECOMPOSE_FUNCTIONAL (pytorch/functorch#814) * add batching rule for block_diag, kill DECOMPOSE_FUNCTIONAL * add batching rule for block_diag, kill DECOMPOSE_FUNCTIONAL