test first step

NCAR · Aug 24, 2023 · f6472ac · f6472ac
1 parent 709a09d
commit f6472ac
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 1 deletion.
diff --git a/include/micm/process/cuda_process_set.hpp b/include/micm/process/cuda_process_set.hpp
@@ -4,7 +4,7 @@
 #pragma once
 
 #include <micm/process/process_set.hpp>
-
+#include <micm/util/cuda_matrix_paran.hpp>
 #ifdef USE_CUDA
 #  include <micm/process/cuda_process_set.cuh>
 #endif
@@ -51,6 +51,7 @@ namespace micm
       const MatrixPolicy<double>& state_variables,
       MatrixPolicy<double>& forcing) const
   {
+    micm::CUDAMatrixParam matrixParam(rate_constants); 
     std::chrono::nanoseconds kernel_duration = micm::cuda::AddForcingTermsKernelDriver(
         rate_constants.AsVector().data(),
         state_variables.AsVector().data(),

diff --git a/include/micm/util/cuda_matrix_param.hpp b/include/micm/util/cuda_matrix_param.hpp
@@ -0,0 +1,13 @@
+#pragma once
+#include <micm/util/vector_matrix.hpp>
+
+namespace micm{
+    class CUDAMatrixParam{
+        public: 
+        double* rate_constants; 
+        template<size_t L>
+        CUDAMatrixParam(const VectorMatrix&<double, L> rateConstants){
+            rate_constants = rateConstants; 
+        }
+    }; 
+}