EAMxx: public methods for cuda in ZonalAvgDiag

Author: mahf708

components/eamxx/src/diagnostics/zonal_avg.hpp

@@ -13,16 +13,6 @@ namespace scream {
 
 class ZonalAvgDiag : public AtmosphereDiagnostic {
 
-  // Utility to compute the contraction of a field along its column dimension.
-  // This is equivalent to f_out = einsum('i,i...k->...k', weight, f_in).
-  // The implementation is such that:
-  // - all Field objects must be allocated
-  // - the first dimension for field, weight, and lat is for the columns (COL)
-  // - the first dimension for result is for the zonal bins (CMP,"bin")
-  // - field and result must be the same dimension, up to 3
-  static void compute_zonal_sum(const Field &result, const Field &field, const Field &weight,
-                                const Field &lat, const ekat::Comm *comm = nullptr);
-
 public:
   // Constructors
   ZonalAvgDiag(const ekat::Comm &comm, const ekat::ParameterList &params);
@@ -37,11 +27,19 @@ class ZonalAvgDiag : public AtmosphereDiagnostic {
 #ifdef KOKKOS_ENABLE_CUDA
 public:
 #endif
+  void initialize_impl(const RunType /*run_type*/);
   void compute_diagnostic_impl();
+  // Utility to compute the contraction of a field along its column dimension.
+  // This is equivalent to f_out = einsum('i,i...k->...k', weight, f_in).
+  // The implementation is such that:
+  // - all Field objects must be allocated
+  // - the first dimension for field, weight, and lat is for the columns (COL)
+  // - the first dimension for result is for the zonal bins (CMP,"bin")
+  // - field and result must be the same dimension, up to 3
+  static void compute_zonal_sum(const Field &result, const Field &field, const Field &weight,
+                                const Field &lat, const ekat::Comm *comm = nullptr);
 
 protected:
-  void initialize_impl(const RunType /*run_type*/);
-
   std::string m_diag_name;
   int m_num_zonal_bins;