def METHOD_NAME(self):
    model = RegressionModel()
    model._original_forward = model.forward
    model.forward = torch.cuda.amp.autocast(dtype=torch.float16)(model.forward)
    model.forward = convert_outputs_to_fp32(model.forward)
    model.forward = torch.compile(model.forward, backend="inductor")
    inputs = torch.randn(4, 10).cuda()
    _ = model(inputs)