cavis/cavis-dnn/cavis-dnn-cudnn/build.gradle

apply from: "${project.rootProject.projectDir}/createTestBackends.gradle"

ext {
    buildTarget = rootProject.ext.buildTarget
}

dependencies {
    implementation platform(projects.cavisCommonPlatform)
    implementation project(":cavis-native:cavis-native-jcublas")
    implementation projects.cavisDnn.cavisDnnApi
    implementation projects.cavisDnn.cavisDnnNn

    implementation group: "org.bytedeco", name: "cuda"
    implementation group: "org.bytedeco", name: "cuda", classifier: buildTarget
    implementation group: "org.bytedeco", name: "cuda", classifier: "${buildTarget}-redist"

    implementation group: "org.bytedeco", name: "javacpp"
    implementation group: "org.bytedeco", name: "javacpp", classifier: buildTarget

    implementation 'com.jakewharton.byteunits:byteunits:0.9.1'

}