diff options
author | Guangda Lai <laigd@google.com> | 2018-07-16 22:56:37 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-07-16 23:01:37 -0700 |
commit | c091185930eee5a3f87cfbe5e367a3cfc8b717e8 (patch) | |
tree | da95c09c4bf1cae335e6dc64735fd993c63ed625 /tensorflow/contrib/tensorrt/resources/trt_allocator.h | |
parent | a013474eae601ae27c406e6c8da34ef762e8f762 (diff) |
Fix various building/testing issues and enable basic TensorRT tests, details are:
1. Add IsGoogleTensorRTEnabled() method and is_tensorrt_enabled() python wrapper to guard trt python tests.
2. Fix nvcc build problems and add corresponding TODOs in some c++ code.
3. Fix various kokoro test config problems (e.g. fix oss build dependencies, add nomac tags for some tests, etc)
PiperOrigin-RevId: 204862004
Diffstat (limited to 'tensorflow/contrib/tensorrt/resources/trt_allocator.h')
-rw-r--r-- | tensorflow/contrib/tensorrt/resources/trt_allocator.h | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/tensorflow/contrib/tensorrt/resources/trt_allocator.h b/tensorflow/contrib/tensorrt/resources/trt_allocator.h index c5d2cec730..97ac82ca5d 100644 --- a/tensorflow/contrib/tensorrt/resources/trt_allocator.h +++ b/tensorflow/contrib/tensorrt/resources/trt_allocator.h @@ -51,6 +51,9 @@ class TRTDeviceAllocator : public nvinfer1::IGpuAllocator { // Allocator implementation wrapping TF device allocators. public: TRTDeviceAllocator(tensorflow::Allocator* allocator); + + // TODO(aaroey): base class doesn't have a virtual destructor, work with + // Nvidia to fix it. virtual ~TRTDeviceAllocator() { VLOG(1) << "Destroying allocator attached to " << allocator_->Name(); } |