С макушки головы, попробуйте переместить
UIGraphicsEndImageContext();
в нижнюю часть функции после сохранения изображения.Кроме того, он выглядит так же, как код, который я использовал.Если это не сработает, можете ли вы предоставить несколько дополнительных деталей о том, как / когда вы вызываете этот метод, и что вы рисуете на экране?