摘要: MindSpore编译构建后Vmap模块的RuntimeError问题 在MindSpore编译计算图的过程中,会把从编译构建好的whl包中引入的模块视为第三方库,也就没有办法在即时编译的阶段入图。普通的math和numpy等第三方库不入图也不会影响计算。但如果是基于MindSpore本身开发的一些函数,如果用到了Jit、Grad和Vmap,那么有可能出现无法入图的问题,就会出现RuntimeError报错。解决方法就是设置一个跟即时编译有关的环境变量,把相关的第三方包引用修改为内部引用。 阅读全文
posted @ 2024-04-11 10:24 DECHIN 阅读(142) 评论(0) 推荐(0) 编辑