def METHOD_NAME(self):
    preds = deepcopy(self.preds)
    target = deepcopy(self.target)
    target_nan = deepcopy(self.target_nan)
    # Regular loss
    loss_true = MSELoss()(preds, target)
    loss_ipu = MSELossIPU()(preds, target)
    self.assertFalse(loss_true.isnan(), "Regular MSELoss is NaN")
    self.assertAlmostEqual(
        loss_true.item(), loss_ipu.item(), places=6, msg="Regular MSELoss is different"
    )
    # Regular loss with NaNs in target
    not_nan = ~target_nan.isnan()
    loss_true = MSELoss()(preds[not_nan], target[not_nan])
    loss_ipu = MSELossIPU()(preds, target_nan)
    self.assertFalse(loss_true.isnan(), "Regular MSELoss with target_nan is NaN")
    self.assertFalse(loss_ipu.isnan(), "Regular MSELossIPU with target_nan is NaN")
    self.assertAlmostEqual(
        loss_true.item(), loss_ipu.item(), places=6, msg="Regular MSELoss with NaN is different"
    )