change to cudaMalloc

shouxieai · Sep 14, 2021 · 7ff2f71 · 7ff2f71
1 parent a68ffed
commit 7ff2f71
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/tensorRT/common/trt_tensor.cpp b/src/tensorRT/common/trt_tensor.cpp
@@ -66,7 +66,7 @@ namespace TRT{
 			release_gpu();
 
 			gpu_size_ = size;
-			checkCudaRuntime(cudaMallocManaged(&gpu_, size));
+			checkCudaRuntime(cudaMalloc(&gpu_, size));
 			checkCudaRuntime(cudaMemset(gpu_, 0, size));
 		}
 		return gpu_;