@ocots to accommodate current issue on GPU https://github.com/MadNLP/MadNLP.jl/issues/518#issue-3664764171