## On this page

  

 

 # api\_cuda\_particles

 Last update: 17.07.2025 

<a id="l00001" name="l00001"></a> 1\#pragma once

<a id="l00002" name="l00002"></a> 2 

<a id="l00003" name="l00003"></a> 3\#include &lt;rocky20/api/device/api\_device\_particles.hpp&gt;

<a id="l00004" name="l00004"></a> 4\#include &lt;rocky20/api/device/api\_cuda.hpp&gt;

<a id="l00005" name="l00005"></a> 5 

<a id="l00006" name="l00006"></a> 6namespace rocky::cuda

<a id="l00007" name="l00007"></a> 7{

<a id="l00008" name="l00008"></a> 8 

<a id="l00009" name="l00009"></a> 9template &lt;typename Callable&gt;

<a id="l00010" name="l00010"></a> 10void dispatch_particles_kernel(

<a id="l00011" name="l00011"></a> 11 [IRockyModel](structIRockyModel.xhtml) &amp;model,

<a id="l00012" name="l00012"></a> 12 Callable h_particle_f_ptr,

<a id="l00013" name="l00013"></a> 13 void **udf_devices_data

<a id="l00014" name="l00014"></a> 14)

<a id="l00015" name="l00015"></a> 15{

<a id="l00016" name="l00016"></a> 16 if (model.[get\_number\_of\_active\_particles](structIRockyModel.xhtml#a38a15145f4f2bb47b79ab785f50d8277)() &lt;= 0)

<a id="l00017" name="l00017"></a> 17 return;

<a id="l00018" name="l00018"></a> 18 

<a id="l00019" name="l00019"></a> 19 for (int i=0; i&lt;model.[get\_number\_of\_devices](structIRockyModel.xhtml#a1fd1011aa809fe64647e9cacf1956a11)(); i++)

<a id="l00020" name="l00020"></a> 20 {

<a id="l00021" name="l00021"></a> 21 CUDA_SET_DEVICE(model.[get\_device\_id](structIRockyModel.xhtml#ae8f8865c809a31988291f9180d7cda2b)(i));

<a id="l00022" name="l00022"></a> 22 

<a id="l00023" name="l00023"></a> 23 void *udf_device_data = udf_devices_data[i];

<a id="l00024" name="l00024"></a> 24 device::execute_particles_function&lt;CudaBackend&gt;&lt;&lt;&lt;CD_NBLOCKS, CD_NTHREADS&gt;&gt;&gt;(

<a id="l00025" name="l00025"></a> 25 model.get_device_data(i), model.[get\_current\_time](structIRockyModel.xhtml#a93ef35ae1eb9cf50ca77ab2070a769ab)(),

<a id="l00026" name="l00026"></a> 26 h_particle_f_ptr, udf_device_data

<a id="l00027" name="l00027"></a> 27 );

<a id="l00028" name="l00028"></a> 28 }

<a id="l00029" name="l00029"></a> 29 

<a id="l00030" name="l00030"></a> 30 // Sync all GPUs

<a id="l00031" name="l00031"></a> 31 sync_devices(model);

<a id="l00032" name="l00032"></a> 32}

<a id="l00033" name="l00033"></a> 33 

<a id="l00034" name="l00034"></a> 34template &lt;typename Callable&gt;

<a id="l00035" name="l00035"></a> 35__global__ void execute_instantaneous_breakage_function_kernel(

<a id="l00036" name="l00036"></a> 36 SDeviceModel* device_data, int device_id,

<a id="l00037" name="l00037"></a> 37 Callable instantaneous_breakage_f_ptr,

<a id="l00038" name="l00038"></a> 38 void* udf_device_data,

<a id="l00039" name="l00039"></a> 39 cu_real time_curr

<a id="l00040" name="l00040"></a> 40)

<a id="l00041" name="l00041"></a> 41{

<a id="l00042" name="l00042"></a> 42 device::compute_instantaneous_breakage&lt;CudaBackend&gt;(

<a id="l00043" name="l00043"></a> 43 device_data, device_id,

<a id="l00044" name="l00044"></a> 44 instantaneous_breakage_f_ptr, udf_device_data,

<a id="l00045" name="l00045"></a> 45 time_curr

<a id="l00046" name="l00046"></a> 46 );

<a id="l00047" name="l00047"></a> 47}

<a id="l00048" name="l00048"></a> 48 

<a id="l00049" name="l00049"></a> 49template &lt;typename Callable&gt;

<a id="l00050" name="l00050"></a> 50void dispatch_instantaneous_breakage_kernel(

<a id="l00051" name="l00051"></a> 51 [IRockyModel](structIRockyModel.xhtml)&amp; model,

<a id="l00052" name="l00052"></a> 52 Callable h_instantaneous_breakage_f_ptr,

<a id="l00053" name="l00053"></a> 53 void** udf_devices_data

<a id="l00054" name="l00054"></a> 54)

<a id="l00055" name="l00055"></a> 55{

<a id="l00056" name="l00056"></a> 56 for (int i = 0; i &lt; model.[get\_number\_of\_devices](structIRockyModel.xhtml#a1fd1011aa809fe64647e9cacf1956a11)(); i++)

<a id="l00057" name="l00057"></a> 57 {

<a id="l00058" name="l00058"></a> 58 void* udf_device_data = udf_devices_data[i];

<a id="l00059" name="l00059"></a> 59 SDeviceModel* device_data = model.get_device_data(i);

<a id="l00060" name="l00060"></a> 60 

<a id="l00061" name="l00061"></a> 61 CUDA_SET_DEVICE(model.[get\_device\_id](structIRockyModel.xhtml#ae8f8865c809a31988291f9180d7cda2b)(i));

<a id="l00062" name="l00062"></a> 62 rocky::cuda::execute_instantaneous_breakage_function_kernel&lt;&lt;&lt;CD_NBLOCKS, CD_NTHREADS&gt;&gt;&gt;(

<a id="l00063" name="l00063"></a> 63 device_data,

<a id="l00064" name="l00064"></a> 64 i,

<a id="l00065" name="l00065"></a> 65 h_instantaneous_breakage_f_ptr,

<a id="l00066" name="l00066"></a> 66 udf_device_data,

<a id="l00067" name="l00067"></a> 67 model._impl-&gt;iteration.current_time);

<a id="l00068" name="l00068"></a> 68 }

<a id="l00069" name="l00069"></a> 69 

<a id="l00070" name="l00070"></a> 70 //Sync all GPUs

<a id="l00071" name="l00071"></a> 71 sync_devices(model);

<a id="l00072" name="l00072"></a> 72}

<a id="l00073" name="l00073"></a> 73 

<a id="l00074" name="l00074"></a> 74} // namespace rocky::cuda

[IRockyModel](structIRockyModel.xhtml)

**Definition** rocky_api.h:60



[IRockyModel::get\_number\_of\_devices](structIRockyModel.xhtml#a1fd1011aa809fe64647e9cacf1956a11)

int get_number_of_devices() const

**Definition** rocky_api.cpp:159



[IRockyModel::get\_number\_of\_active\_particles](structIRockyModel.xhtml#a38a15145f4f2bb47b79ab785f50d8277)

int get_number_of_active_particles() const

**Definition** rocky_api.cpp:139



[IRockyModel::get\_current\_time](structIRockyModel.xhtml#a93ef35ae1eb9cf50ca77ab2070a769ab)

double get_current_time() const

**Definition** rocky_api.cpp:129



[IRockyModel::get\_device\_id](structIRockyModel.xhtml#ae8f8865c809a31988291f9180d7cda2b)

int get_device_id(int device_index) const

**Definition** rocky_api.cpp:164