Swift - 圆形边缘仅在4“设备上围绕2个边缘延伸

Question

我正在使用圆边扩展的应用程序。它工作正常，除了4“设备（见图）我该怎么办？

谢谢！

这是我的代码

extension UIView {
    func roundCorners(corners:UIRectCorner, radius: CGFloat) {
        let path = UIBezierPath(roundedRect: self.bounds, byRoundingCorners: corners, cornerRadii: CGSize(width: radius, height: radius))
        let mask = CAShapeLayer()
        mask.path = path.cgPath
        self.layer.mask = mask
    }
}