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

23 lines
802 B
Groovy

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'
}