change to cudaMalloc

Wish · Wish · commit 7ff2f7180e27 · 2021-09-14T11:11:57.000+08:00
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_;

Original file line number	Diff line number	Diff line change
`@@ -66,7 +66,7 @@ namespace TRT{`
`66`	`66`	`release_gpu();`
`67`	`67`
`68`	`68`	`gpu_size_ = size;`
`69`		`- checkCudaRuntime(cudaMallocManaged(&gpu_, size));`
	`69`	`+ checkCudaRuntime(cudaMalloc(&gpu_, size));`
`70`	`70`	`checkCudaRuntime(cudaMemset(gpu_, 0, size));`
`71`	`71`	`}`
`72`	`72`	`return gpu_;`