Skip to content

Commit 2656e90

Browse files
authored
fix a shape bug in anakin-nvgpu, test=release/1.5 (#19383)
1 parent c328a9e commit 2656e90

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

paddle/fluid/inference/api/api_anakin_engine.cc

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -195,7 +195,7 @@ bool PaddleInferenceAnakinPredictor<T, P, R>::RunImpl(
195195
<< "'s type is not float";
196196
}
197197
auto d_tensor_p = this->executor_p_->get_in(input.name);
198-
auto net_shape = d_tensor_p->shape();
198+
auto net_shape = d_tensor_p->valid_shape();
199199
if (net_shape.size() != input.shape.size()) {
200200
LOG(FATAL) << " input " << input.name
201201
<< "'s shape size should be equal to that of net";

0 commit comments

Comments
 (0)