Hi,
Thanks for the report.
Could you give some details about:

Which GPU, cuda version and pytorch version you’re using?
What are the y_hat and t Tensor you give as input. In particular could you share for both t.size(), t.stride() and t.storage_offset()?

cc @ptrblck do you know what could be causing this?