From 66acebf125cdd77f5df6e75da9a8584865216432 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E8=96=9B=E9=B9=8F?= Date: Mon, 13 Feb 2023 00:54:41 +0000 Subject: [PATCH] =?UTF-8?q?=E5=9B=9E=E9=80=80=20'Pull=20Request=20!1994=20?= =?UTF-8?q?:=20=E4=BF=AE=E6=94=B9TF2=E4=B8=8A=E5=8D=95=E7=AE=97=E5=AD=90?= =?UTF-8?q?=E9=BB=98=E8=AE=A4=E5=9C=A8CPU=E6=89=A7=E8=A1=8C'?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../npu_device/core/op_executors/npu_static_shape_op.cpp | 8 +------- 1 file changed, 1 insertion(+), 7 deletions(-) diff --git a/tf_adapter_2.x/npu_device/core/op_executors/npu_static_shape_op.cpp b/tf_adapter_2.x/npu_device/core/op_executors/npu_static_shape_op.cpp index f3c58b233..84668f737 100644 --- a/tf_adapter_2.x/npu_device/core/op_executors/npu_static_shape_op.cpp +++ b/tf_adapter_2.x/npu_device/core/op_executors/npu_static_shape_op.cpp @@ -39,16 +39,10 @@ std::string NpuStaticShapeOp::AttachedDebugString() const { return ss.str(); } -#ifdef TFA2_COMPILED_FOR_EAGER_MODE -#define TFA2_EAGER_ENABLED true -#else -#define TFA2_EAGER_ENABLED false -#endif - void NpuStaticShapeOp::RunWithShape(TFE_Context *context, NpuDevice *device, const OpExecutor *spec, TensorShapes output_shapes, int num_inputs, TFE_TensorHandle **inputs, int num_outputs, TFE_TensorHandle **outputs, TF_Status *status) { - if (kGraphEngineGreedyMemory || (!TFA2_EAGER_ENABLED)) { + if (kGraphEngineGreedyMemory) { DLOG() << "NPU Executing op " << spec->Op() << " fallback cpu in graph engine greedy memory mode"; device->FallbackCPU(context, spec->NodeDef(), num_inputs, inputs, num_outputs, outputs, status); return; -- Gitee