This works now following way:

asciidoctorj {
    modules {
        diagram.use()
    }
}

See: https://asciidoctor.github.io/asciidoctor-gradle-plugin/master/user-guide/#diagram