cuda Assertion failed when not compiling the model

Question

我们在 Cuda 中有一个错误说

Assertion `srcIndex < srcSelectDimSize` failed

我们找不到原因。只有在模型上使用 torch.compile 时，它才消除了故障。

             self.model = torch.compile(self.model).to(self.device)

为什么这可以解决断言失败的任何想法？我们想了解一下原来的失败

感谢任何想法 里尔