pytorch

5cc258ec - make block_diag composite compliant

make block_diag composite compliant Pull Request resolved: https://github.com/pytorch/pytorch/pull/77716 Approved by: https://github.com/zou3519