Keep logger alive to avoid segfault in trt_engine_resource_ops test
diff --git a/tensorflow/compiler/tf2tensorrt/kernels/trt_engine_resource_ops_test.cc b/tensorflow/compiler/tf2tensorrt/kernels/trt_engine_resource_ops_test.cc
index 4d8f0ec..f2bd75d 100644
--- a/tensorflow/compiler/tf2tensorrt/kernels/trt_engine_resource_ops_test.cc
+++ b/tensorflow/compiler/tf2tensorrt/kernels/trt_engine_resource_ops_test.cc
@@ -55,9 +55,8 @@
}
TrtUniquePtrType<nvinfer1::ICudaEngine> CreateTRTEngine() {
- Logger logger;
TrtUniquePtrType<nvinfer1::IBuilder> builder(
- nvinfer1::createInferBuilder(logger));
+ nvinfer1::createInferBuilder(logger_));
TrtUniquePtrType<nvinfer1::INetworkDefinition> network(
builder->createNetwork());
@@ -87,6 +86,7 @@
EXPECT_NE(nullptr, engine);
return engine;
}
+ Logger logger_;
};
TEST_F(TRTEngineResourceOpsTest, Basic) {