Free light cache on host after DAG gen, and add warning of slow DAG gen

ethereum-mining · Jul 28, 2020 · 09bf291 · 09bf291
1 parent 4da38fc
commit 09bf291
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 0 deletions.
diff --git a/libethash-cl/CLMiner.cpp b/libethash-cl/CLMiner.cpp
@@ -875,8 +875,11 @@ bool CLMiner::initEpoch_internal()
                     throw;
             }
             if (light_on_host)
+            {
                 m_light.emplace_back(m_context[0], CL_MEM_READ_ONLY | CL_MEM_ALLOC_HOST_PTR,
                     m_epochContext.lightSize);
+                cllog << "WARNING: Generating DAG will take minutes, not seconds";
+            }
             cllog << "Loading kernels";
 
             // If we have a binary kernel to use, let's try it

diff --git a/libethash-cuda/CUDAMiner.cpp b/libethash-cuda/CUDAMiner.cpp
@@ -126,8 +126,11 @@ bool CUDAMiner::initEpoch_internal()
 
             // create buffer for cache
             if (lightOnHost)
+            {
                 CUDA_SAFE_CALL(cudaHostAlloc(reinterpret_cast<void**>(&light),
                     m_epochContext.lightSize, cudaHostAllocDefault));
+                cudalog << "WARNING: Generating DAG will take minutes, not seconds";
+            }
             else
                 CUDA_SAFE_CALL(
                     cudaMalloc(reinterpret_cast<void**>(&light), m_epochContext.lightSize));
@@ -158,6 +161,9 @@ bool CUDAMiner::initEpoch_internal()
         ethash_generate_dag(
             m_epochContext.dagSize, m_settings.gridSize, m_settings.blockSize, m_streams[0]);
 
+        if (lightOnHost)
+            CUDA_SAFE_CALL(cudaFreeHost(reinterpret_cast<void*>(light)));
+
         cudalog << "Generated DAG + Light in "
                 << std::chrono::duration_cast<std::chrono::milliseconds>(
                        std::chrono::steady_clock::now() - startInit)