Skip to content

Commit 05e4be6

Browse files
committed
fix the crash when input variable is bool type, test=develop
1 parent ac33c0c commit 05e4be6

File tree

2 files changed

+4
-3
lines changed

2 files changed

+4
-3
lines changed

python/paddle/fluid/io.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1378,7 +1378,7 @@ def save_inference_model(dirname,
13781378
with program_guard(main_program):
13791379
uniq_target_vars = []
13801380
for i, var in enumerate(target_vars):
1381-
if isinstance(var, Variable):
1381+
if isinstance(var, Variable) and var.dtype != paddle.bool:
13821382
var = layers.scale(
13831383
var, 1., name="save_infer_model/scale_{}".format(i))
13841384
uniq_target_vars.append(var)

python/paddle/static/io.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -192,8 +192,9 @@ def normalize_program(program, feed_vars, fetch_vars):
192192
with program_guard(program):
193193
uniq_fetch_vars = []
194194
for i, var in enumerate(fetch_vars):
195-
var = layers.scale(
196-
var, 1., name="save_infer_model/scale_{}".format(i))
195+
if var.dtype != paddle.bool:
196+
var = layers.scale(
197+
var, 1., name="save_infer_model/scale_{}".format(i))
197198
uniq_fetch_vars.append(var)
198199
fetch_vars = uniq_fetch_vars
199200

0 commit comments

Comments
 (0)