Changeset 2645 for XIOS3/dev

Timestamp:

07/24/24 16:24:21 (7 months ago)

Author:

jderouillat

Message:

Introduce Kokkos::parallel_for in local connector and, in model to client filter

Location:

XIOS3/dev/XIOS_KOKKOS/src

Files:

• distribution/local_connector.hpp (modified) (1 diff)
• filter/model_to_client_source_filter.cpp (modified) (1 diff)

XIOS3/dev/XIOS_KOKKOS/src/distribution/local_connector.hpp

@@ -108 +108 @@
           Kokkos::View<T*,Kokkos::DefaultExecutionSpace::memory_space, Kokkos::MemoryTraits<Kokkos::Unmanaged> > k_input ( (T*)input, getSrcSize() );
           Kokkos::View<T*,Kokkos::DefaultExecutionSpace::memory_space, Kokkos::MemoryTraits<Kokkos::Unmanaged> > k_output( output, getDstSize() );
-          for(int i=0;i<mask_.size();i++) k_output(i)=missingValue ;
-          for(int i=0;i<size;i++)
+
+          Kokkos::parallel_for( "local_connector_missing", mask_.size(), KOKKOS_LAMBDA( const size_t i )
+          {
+            k_output(i)=missingValue ;
+          });
+          Kokkos::fence();
+          Kokkos::parallel_for( "local_connector_transfer", size, KOKKOS_LAMBDA( const size_t i )
           {
             k_output(connector_out(i))=k_input(connector_in(i)) ;
-          }
+          });
+          Kokkos::fence();
 
           // - //

XIOS3/dev/XIOS_KOKKOS/src/filter/model_to_client_source_filter.cpp

@@ -49 +49 @@
         const double nanValue = std::numeric_limits<double>::quiet_NaN();
         const size_t nbData = packet->data.numElements();
-        for (size_t idx = 0; idx < nbData; ++idx)
-          if (defaultValue_ == packet->data(idx))   packet->data(idx) = nanValue;
+        auto packetData = packet->data;
+        Kokkos::parallel_for( "filter_missing_value", nbData, KOKKOS_LAMBDA( const size_t idx )
+        {
+          if (defaultValue_ == packetData(idx))   packetData(idx) = nanValue;
+        });
+        Kokkos::fence();
     }