UoB-HPC / BabelStream

STREAM, for lots of devices written in many programming models
Other
313 stars 109 forks source link

set device_id for Kokkos #184

Closed milthorpe closed 5 months ago

milthorpe commented 5 months ago

Previously, the Kokkos implementation ignores the command line parameter --device and always runs using the default device. This change initializes Kokkos with the specified device.

tomdeakin commented 5 months ago

Looks good, just re-running the CI.

tomdeakin commented 5 months ago

CI is failing only after the cpp test passes, so this looks good to merge. Thanks for the contribution!