numerical stability problem

[wuzirui]

Hi! I've recently run into a numerical stability problem with the rasterization kernel as shown below: I call the rasterization kernel with the following command: I've checked the inputs and all the tensors are guaranteed to be not nan, but when I tried to calculate the gradient of render_outputs or alpha over gaussian params (e.g. scales), nan values appear:

Could you please give me some hints on which part of the code may be numerically unstable so I can come up with some walkaround, or where I shall start to fix the problem? Thanks a lot!