/
build.gradle
61 lines (52 loc) · 1.98 KB
/
build.gradle
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
plugins {
id 'application'
id 'jacoco'
}
dependencies {
implementation "commons-cli:commons-cli:${commons_cli_version}"
implementation "org.apache.logging.log4j:log4j-slf4j-impl:${log4j_slf4j_version}"
implementation project(":basicdataset")
implementation project(":model-zoo")
implementation project(":testing")
runtimeOnly project(":engines:mxnet:mxnet-model-zoo")
runtimeOnly project(":engines:pytorch:pytorch-model-zoo")
runtimeOnly project(":engines:pytorch:pytorch-jni")
runtimeOnly project(":engines:tensorflow:tensorflow-model-zoo")
runtimeOnly project(":engines:ml:xgboost")
runtimeOnly project(":engines:ml:lightgbm")
runtimeOnly project(":engines:onnxruntime:onnxruntime-engine")
if (System.getProperty("ai.djl.default_engine") == "TFLite") {
runtimeOnly project(":engines:tflite:tflite-engine")
} else if (System.getProperty("ai.djl.default_engine") == "PaddlePaddle") {
runtimeOnly project(":engines:paddlepaddle:paddlepaddle-model-zoo")
}
}
java {
compileJava {
sourceCompatibility = JavaVersion.VERSION_11
targetCompatibility = JavaVersion.VERSION_11
options.compilerArgs.removeAll(["--release", "8"])
}
}
tasks.register('copyDependencies', Copy) {
into "build/dependencies"
from configurations.runtimeClasspath
}
application {
mainClass = System.getProperty("main", "ai.djl.integration.IntegrationTest")
}
run {
environment("TF_CPP_MIN_LOG_LEVEL", "1") // turn off TensorFlow print out
systemProperties System.getProperties()
systemProperties.remove("user.dir")
systemProperty("file.encoding", "UTF-8")
jvmArgs "-Xverify:none"
}
tasks.register('debugEnv', JavaExec) {
classpath = sourceSets.main.runtimeClasspath
systemProperties System.getProperties()
systemProperties.remove("user.dir")
systemProperties.put("ai.djl.logging.level", "debug")
mainClass = "ai.djl.integration.util.DebugEnvironment"
}
tasks.distTar.enabled = false