yjsunnn

Thanks. It may be my mistake and I will check again. Sorry for this mistake.

What's your pytorch version? When my torch version==1.9.0, I got the same MAGMA error. And I fixed the problem by downgrading pytorch to 1.7 version.