## On this page

  

 

 # api\_cuda

 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/cuda/cuda\_backend.hpp&gt;

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

<a id="l00005" name="l00005"></a> 5\#include &lt;rocky20/api/rocky\_api.h&gt;

<a id="l00006" name="l00006"></a> 6 

<a id="l00007" name="l00007"></a> 7namespace rocky::cuda

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

<a id="l00009" name="l00009"></a> 9 

<a id="l00010" name="l00010"></a> 10inline void sync_devices([IRockyModel](structIRockyModel.xhtml) &amp;model)

<a id="l00011" name="l00011"></a> 11{

<a id="l00012" name="l00012"></a> 12 // Sync all GPUs

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

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

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

<a id="l00016" name="l00016"></a> 16 CUDA_ERROR_CHECK(cudaDeviceSynchronize());

<a id="l00017" name="l00017"></a> 17 }

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

<a id="l00019" name="l00019"></a> 19 

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

[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\_device\_id](structIRockyModel.xhtml#ae8f8865c809a31988291f9180d7cda2b)

int get_device_id(int device_index) const

**Definition** rocky_api.cpp:164