device_context.cc.patch 1.3 KB
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46
diff --git a/paddle/fluid/platform/device_context.cc b/paddle/fluid/platform/device_context.cc
index 904e4854ba..4f8c955d8c 100644
--- a/paddle/fluid/platform/device_context.cc
+++ b/paddle/fluid/platform/device_context.cc
@@ -466,15 +466,15 @@ CUDAContext::CUDAContext(const CUDAPlace& place,
   place_ = place;
   CUDADeviceGuard guard(place_.device);
   stream_.reset(new stream::CUDAStream(place, priority, flag));
-  InitEigenContext();
-  InitCuBlasContext();
-  InitCuDNNContext();
+  //InitEigenContext();
+  //InitCuBlasContext();
+  //InitCuDNNContext();
 #ifndef PADDLE_WITH_HIP
 #if CUDA_VERSION >= 11060
-  InitCuBlasLtContext();
+  //InitCuBlasLtContext();
 #endif
-  InitCuSparseContext();
-  InitCuSolverContext();
+  //InitCuSparseContext();
+  //InitCuSolverContext();
 #endif
 }
 
@@ -506,14 +506,14 @@ void CUDAContext::SetStream(gpuStream_t stream) {
 
 CUDAContext::~CUDAContext() {
   CUDADeviceGuard guard(place_.device);
-  DestoryCuDNNContext();
-  DestoryCuBlasContext();
+  //DestoryCuDNNContext();
+  //DestoryCuBlasContext();
 #ifndef PADDLE_WITH_HIP
 #if CUDA_VERSION >= 11060
-  InitCuBlasLtContext();
+  //InitCuBlasLtContext();
 #endif
-  DestoryCuSparseContext();
-  DestoryCuSolverContext();
+  //DestoryCuSparseContext();
+  //DestoryCuSolverContext();
 #endif
 }