Skip to content

Commit a568240

Browse files
yinggehmc-nv
andauthored
ci: Fix L0_infer--cpu* tests (#8123)
Co-authored-by: Misha Chornyi <99709299+mc-nv@users.noreply.github.com>
1 parent 7337c13 commit a568240

File tree

2 files changed

+8
-8
lines changed

2 files changed

+8
-8
lines changed

qa/L0_infer/test.sh

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -238,7 +238,7 @@ function generate_model_repository() {
238238

239239
ENSEMBLE_MODELS="${ENSEMBLE_MODELS} batch_to_nobatch_float32_float32_float32 batch_to_nobatch_nobatch_float32_float32_float32 nobatch_to_batch_float32_float32_float32 nobatch_to_batch_nobatch_float32_float32_float32 mix_nobatch_batch_float32_float32_float32"
240240

241-
if [[ $BACKENDS == *"plan"* ]] ; then
241+
if [[ $BACKENDS == *"libtorch"* ]] ; then
242242
ENSEMBLE_MODELS="${ENSEMBLE_MODELS} mix_platform_float32_float32_float32 mix_ensemble_int32_float32_float32"
243243
fi
244244

qa/ensemble_models/mix_platform_float32_float32_float32/config.pbtxt

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -70,7 +70,7 @@ ensemble_scheduling {
7070
}
7171
},
7272
{
73-
model_name: "plan_float32_float32_float32"
73+
model_name: "libtorch_float32_float32_float32"
7474
model_version: 1
7575
input_map {
7676
key: "INPUT0"
@@ -81,12 +81,12 @@ ensemble_scheduling {
8181
value: "INPUT1"
8282
}
8383
output_map {
84-
key: "OUTPUT0"
84+
key: "OUTPUT__0"
8585
value: "double_input1"
8686
}
8787
},
8888
{
89-
model_name: "plan_float32_float32_float32"
89+
model_name: "libtorch_float32_float32_float32"
9090
model_version: 1
9191
input_map {
9292
key: "INPUT0"
@@ -97,7 +97,7 @@ ensemble_scheduling {
9797
value: "INPUT0"
9898
}
9999
output_map {
100-
key: "OUTPUT1"
100+
key: "OUTPUT__1"
101101
value: "input0_val"
102102
}
103103
},
@@ -118,7 +118,7 @@ ensemble_scheduling {
118118
}
119119
},
120120
{
121-
model_name: "plan_float32_float32_float32"
121+
model_name: "libtorch_float32_float32_float32"
122122
model_version: 1
123123
input_map {
124124
key: "INPUT0"
@@ -129,11 +129,11 @@ ensemble_scheduling {
129129
value: "input1_val"
130130
}
131131
output_map {
132-
key: "OUTPUT0"
132+
key: "OUTPUT__0"
133133
value: "OUTPUT0"
134134
}
135135
output_map {
136-
key: "OUTPUT1"
136+
key: "OUTPUT__1"
137137
value: "OUTPUT1"
138138
}
139139
}

0 commit comments

Comments
 (0)