Hola, si si funciona con todas las notificaciones ya sea con la pantalla prendida o apaga, para activarlo ve a:
Ajustes, Notificaciones, Aplicaciones incluidas y seleccionas todas o las que quieras usar. Para configurar el efecto edge, vas a Ajustes de resumen emergente, Estilo de iluminación. También ahi dentro puedes activar la opción de ver las notificaciones con pantalla apagada.
Espero haberte ayudado :) de ser así te agradecería marcar mi respuesta como solución en los tres puntitos de aquí arriba.
