Commit 5a3a4256 authored by Andreas Marek's avatar Andreas Marek
Browse files

Pinned memory in elpa2_bandred

parent 43fdb47d
......@@ -290,6 +290,11 @@
successCUDA = cuda_malloc(a_dev, lda*na_cols* size_of_datatype)
check_alloc_cuda("bandred: a_dev", successCUDA)
successCUDA = cuda_host_register(int(loc(vav),kind=c_intptr_t), &
nbw * nbw * size_of_datatype,&
cudaHostRegisterDefault)
check_host_register_cuda("bandred: vav", successCUDA)
successCUDA = cuda_malloc(vav_dev, nbw*nbw* size_of_datatype)
check_alloc_cuda("bandred: vav_dev", successCUDA)
endif ! useGPU
......@@ -1567,6 +1572,9 @@
successCUDA = cuda_free(tmat_dev)
check_dealloc_cuda("bandred: tmat_dev ", successCUDA)
successCUDA = cuda_host_unregister(int(loc(vav),kind=c_intptr_t))
check_host_unregister_cuda("bandred: vav", successCUDA)
if (associated(umcCUDA)) then
nullify(umcCUDA)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment