!46221 reduce类算子,动态shape场景下,kernel_mod的all_match_、need_skip_execute未置为默认值;kernel_mod复用场景可能出现问题。
Merge pull request !46221 from huoxinyou/1130reducegpu
This commit is contained in:
commit
636e62d371
|
@ -179,6 +179,8 @@ void ArrayReduceGpuKernelMod::InitCudnnResource() {
|
|||
int ArrayReduceGpuKernelMod::Resize(const BaseOperatorPtr &base_operator, const std::vector<KernelTensorPtr> &inputs,
|
||||
const std::vector<KernelTensorPtr> &outputs,
|
||||
const std::map<uint32_t, tensor::TensorPtr> &inputsOnHost) {
|
||||
need_skip_execute_ = false;
|
||||
all_match_ = false;
|
||||
int ret = KernelMod::Resize(base_operator, inputs, outputs, inputsOnHost);
|
||||
if (ret != KRET_OK) {
|
||||
InitCudnnResource();
|
||||
|
|
Loading…
Reference in New Issue