我们用MindIE做基础模型的推理速度是挺快的，但是我们在对模型结构进行修改后，无法使用MindIE做推理，因为MindIE里面从token id到logits值是闭源的一个完整算子，我们通过什么办法可以修改这里面的模型结构呢？