Skip to content

Instantly share code, notes, and snippets.

@DavidPoliakoff
Created March 12, 2021 19:41
Show Gist options
  • Save DavidPoliakoff/e80aefe8198061e9b9247de373cea60c to your computer and use it in GitHub Desktop.
Save DavidPoliakoff/e80aefe8198061e9b9247de373cea60c to your computer and use it in GitHub Desktop.
Kokkos Tool
extern "C" void kokkosp_begin_parallel_for(const char* name, const uint32_t devID, uint64_t* kID){
if(name matches a regex){
jonathans_api_wrapper_start();
*kID = 1;
}
}
extern "C" void kokkosp_end_parallel_for(uint64_t kID){
if(kID==1){
jonathans_api_wrapper_stop();
}
}
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment