fix(pt): Treat cuBLAS allocation failures as PyTorch OOM during auto bs#5443

njzjz merged 1 commit into

deepmodeling:masterdeepmodeling/deepmd-kit:masterfrom

OutisLi:pr/bsOutisLi/dpmd-public:pr/bsCopy head branch name to clipboard