program:tensorflow - prune lib-tensorflow by benchmark_[cpu|cuda] cmd.
{
"backup_data_uid": "c0bacdae0ce2b532",
- "compile_deps": {
- "lib-tensorflow": {
- "local": "yes",
- "name": "TensorFlow library",
- "skip_pruning_by_other_deps": "yes",
- "sort": 10,
- "tags": "lib,tensorflow"
- }
- },
"data_name": "tensorflow",
- "process_in_tmp": "yes",
- "program": "yes",
"no_compile": "yes",
"no_target_file": "yes",
+ "process_in_tmp": "yes",
+ "program": "yes",
"run_cmds": {
- "TensorFlow_benchmark": {
+ "benchmark_cpu": {
"dataset_tags": [
"benchmark",
"tensorflow",
"ignore_return_code": "no",
"run_time": {
"run_cmd_main": "$<<CK_ENV_COMPILER_PYTHON_FILE>>$ $#dataset_path#$$#dataset_filename#$ --batch_size=$<<BATCH_SIZE>>$ --num_batches=$<<NUM_BATCHES>>$"
+ },
+ "update_deps": {
+ "lib-tensorflow": {
+ "no_tags": "tensorflow-cuda",
+ "tags": "tensorflow-cpu"
+ }
+ }
+ },
+ "benchmark_cuda": {
+ "dataset_tags": [
+ "benchmark",
+ "tensorflow",
+ "dataset"
+ ],
+ "ignore_return_code": "no",
+ "run_time": {
+ "run_cmd_main": "$<<CK_ENV_COMPILER_PYTHON_FILE>>$ $#dataset_path#$$#dataset_filename#$ --batch_size=$<<BATCH_SIZE>>$ --num_batches=$<<NUM_BATCHES>>$"
+ },
+ "update_deps": {
+ "lib-tensorflow": {
+ "no_tags": "tensorflow-cpu",
+ "tags": "tensorflow-cuda"
+ }
}
}
},
"BATCH_SIZE": 32,
"NUM_BATCHES": 25
},
+ "run_deps": {
+ "lib-tensorflow": {
+ "local": "yes",
+ "name": "TensorFlow library",
+ "skip_pruning_by_other_deps": "no",
+ "sort": 10,
+ "tags": "lib,tensorflow"
+ }
+ },
"tags": [
+ "benchmark",
"tensorflow",
"demo"
]
