mirror of
https://github.com/PaddlePaddle/FastDeploy.git
synced 2026-04-23 00:17:25 +08:00
temporary fix bug of 0 size tensor (#4844)
This commit is contained in:
@@ -2016,6 +2016,10 @@ class GPUModelRunner(ModelRunnerBase):
|
||||
if not self.not_need_stop():
|
||||
self._execute_empty_input()
|
||||
return None
|
||||
# TODO: Fix ORC 0 size bug
|
||||
if self.forward_meta.ids_remove_padding.shape[0] == 0:
|
||||
self.share_inputs["not_need_stop"].zero_()
|
||||
return None
|
||||
|
||||
# 2. Padding inputs for cuda graph
|
||||
self.padding_cudagraph_inputs()
|
||||
|
||||
Reference in New Issue
Block a user