pwm: Switch to the atomic API