From 772fa14dbe5760389f1f4fb4b9e876b3798c5333 Mon Sep 17 00:00:00 2001 From: wangxiaoxin-sherie Date: Tue, 17 Dec 2024 19:09:14 +0800 Subject: [PATCH] safty check. --- mx_driving/fused/ops/kernels/op_host/add_relu.cpp | 6 ++++++ .../ops/kernels/op_host/subm_sparse_conv3d_tiling.cpp | 6 ++++++ 2 files changed, 12 insertions(+) diff --git a/mx_driving/fused/ops/kernels/op_host/add_relu.cpp b/mx_driving/fused/ops/kernels/op_host/add_relu.cpp index c90ad849..a2e92050 100644 --- a/mx_driving/fused/ops/kernels/op_host/add_relu.cpp +++ b/mx_driving/fused/ops/kernels/op_host/add_relu.cpp @@ -22,6 +22,9 @@ static int32_t GetCeilInt(int32_t value1, int32_t value2) static ge::graphStatus TilingFunc(gert::TilingContext* context) { AddReluTilingData tiling; + if (context == nullptr) { + return ge::GRAPH_FAILED; + } auto platformInfoptr = context->GetPlatformInfo(); if (platformInfoptr == nullptr) { return ge::GRAPH_FAILED; @@ -53,6 +56,9 @@ static ge::graphStatus TilingFunc(gert::TilingContext* context) tiling.set_last_copy_tail(coreLast % availableUbSize); tiling.set_box_number(totalResult); tiling.set_available_ub_size(availableUbSize); + if (context->GetRawTilingData() == nullptr) { + return ge::GRAPH_FAILED; + } tiling.SaveToBuffer(context->GetRawTilingData()->GetData(), context->GetRawTilingData()->GetCapacity()); context->GetRawTilingData()->SetDataSize(tiling.GetDataSize()); size_t *currentWorkspace = context->GetWorkspaceSizes(1); diff --git a/mx_driving/spconv/ops/kernels/op_host/subm_sparse_conv3d_tiling.cpp b/mx_driving/spconv/ops/kernels/op_host/subm_sparse_conv3d_tiling.cpp index ae9de2b3..c9838c22 100644 --- a/mx_driving/spconv/ops/kernels/op_host/subm_sparse_conv3d_tiling.cpp +++ b/mx_driving/spconv/ops/kernels/op_host/subm_sparse_conv3d_tiling.cpp @@ -19,6 +19,9 @@ static int32_t GetCeilInt(int32_t value1, int32_t value2) static ge::graphStatus TilingFunc(gert::TilingContext* context) { SubmSparseConv3dTilingData tiling; + if (context == nullptr) { + return ge::GRAPH_FAILED; + } auto platformInfoptr = context->GetPlatformInfo(); if (platformInfoptr == nullptr) { return ge::GRAPH_FAILED; @@ -74,6 +77,9 @@ static ge::graphStatus TilingFunc(gert::TilingContext* context) tiling.set_indices_number(indices_shape.GetDim(1)); tiling.set_available_ub_size(available_ub_size); tiling.set_batch_size(batch_size); + if (context->GetRawTilingData() == nullptr) { + return ge::GRAPH_FAILED; + } tiling.SaveToBuffer(context->GetRawTilingData()->GetData(), context->GetRawTilingData()->GetCapacity()); context->GetRawTilingData()->SetDataSize(tiling.GetDataSize()); size_t *currentWorkspace = context->GetWorkspaceSizes(1); -- Gitee