[TensorRT] added culibos (neo-ai#101)

* added culibos * Revert "added culibos" This reverts commit cdb2fde. * disable cublas by default * removed TensorRTExecManager from graphruntime class * added deprecated message for _tensorrt_subgraph_op * added culibos back since dependency is from cudnn * reset deviceapi pointers in destructor * reset static cudadeviceapi when class is reconstructed * fixed sanity * fixed sanity * removed culibos stuff * removed op_args_ Co-authored-by: Ubuntu <[email protected]>
trevor-m · Sep 1, 2020 · 2bfc3f8 · 2bfc3f8
1 parent 57ff9f9
commit 2bfc3f8
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 1 deletion.
diff --git a/cmake/modules/CUDA.cmake b/cmake/modules/CUDA.cmake
@@ -43,7 +43,7 @@ if(USE_CUDA)
     file(GLOB CONTRIB_CUDNN_SRCS src/runtime/contrib/cudnn/*.cc)
     list(APPEND RUNTIME_SRCS ${CONTRIB_CUDNN_SRCS})
     list(APPEND TVM_RUNTIME_LINKER_LIBS ${CUDA_CUDNN_LIBRARY})
-	  include_directories(${USE_CUDNN}/include)
+    include_directories(${USE_CUDNN}/include)
   endif(USE_CUDNN)
 
   if(USE_CUBLAS)

diff --git a/src/runtime/c_runtime_api.cc b/src/runtime/c_runtime_api.cc
@@ -103,6 +103,7 @@ class DeviceAPIManager {
   std::mutex mutex_;
   // constructor
   DeviceAPIManager() { std::fill(api_.begin(), api_.end(), nullptr); }
+  ~DeviceAPIManager() { std::fill(api_.begin(), api_.end(), nullptr); }
   // Global static variable.
   static DeviceAPIManager* Global() {
     static DeviceAPIManager inst;

diff --git a/src/runtime/cuda/cuda_device_api.cc b/src/runtime/cuda/cuda_device_api.cc
@@ -209,6 +209,7 @@ class CUDADeviceAPI final : public DeviceAPI {
 
   static const std::shared_ptr<CUDADeviceAPI>& Global() {
     static std::shared_ptr<CUDADeviceAPI> inst = std::make_shared<CUDADeviceAPI>();
+    if (inst.use_count() == 0) inst = std::make_shared<CUDADeviceAPI>();
     return inst;
   }