Eu tenho costumeUIView classe que renderiza um gradiente no Swift 2. Estou lutando para fazer um gradiente angular para que ele desenhe da parte superior esquerda para a parte inferior direita. Alguém pode me ajudar um pouco?

import UIKit

class GradientView: UIView {

    let gradientLayer = CAGradientLayer()

    override func awakeFromNib() {
        // 1
        self.backgroundColor = ColorPalette.White

        // 2
        gradientLayer.frame = self.bounds

        // 3
        let color1 = ColorPalette.GrdTop.CGColor as CGColorRef
        let color2 = ColorPalette.GrdBottom.CGColor as CGColorRef
        gradientLayer.colors = [color1, color2]

        // 4
        gradientLayer.locations = [0.0, 1.0]

        // 5
        self.layer.addSublayer(gradientLayer)
    }

}

Suspeito que isso deva ser outra coisa, mas seja o que for que eu insira nada muda.

gradientLayer.locations = [0.0, 1.0]