From f27556333313a23412e085915f3d384310383935 Mon Sep 17 00:00:00 2001 From: Zirui Wu Date: Wed, 7 Apr 2021 13:42:54 -0400 Subject: [PATCH] Codec fix, nullptr isn't checked after a dynamic cast --- .../dataset/engine/opt/optional/tensor_op_fusion_pass.cc | 1 + 1 file changed, 1 insertion(+) diff --git a/mindspore/ccsrc/minddata/dataset/engine/opt/optional/tensor_op_fusion_pass.cc b/mindspore/ccsrc/minddata/dataset/engine/opt/optional/tensor_op_fusion_pass.cc index 23e6e7839b..20830d3d41 100644 --- a/mindspore/ccsrc/minddata/dataset/engine/opt/optional/tensor_op_fusion_pass.cc +++ b/mindspore/ccsrc/minddata/dataset/engine/opt/optional/tensor_op_fusion_pass.cc @@ -39,6 +39,7 @@ Status TensorOpFusionPass::Visit(std::shared_ptr node, bool *const modi if (itr != ops.end()) { MS_LOG(WARNING) << "Fusing pre-build Decode and RandomCropResize into one pre-build."; auto op = dynamic_cast((*(itr + 1))->Build().get()); + RETURN_UNEXPECTED_IF_NULL(op); (*itr) = std::make_shared(std::make_shared(*op)); ops.erase(itr + 1); node->setOperations(ops);