run_on_device.hpp

/* VecMem project, part of the ACTS project (R&D line)
 *
 * (c) 2021 CERN for the benefit of the ACTS project
 *
 * Mozilla Public License Version 2.0
 */
#pragma once
 
// Local include(s).
#include "hip_error_handling.hpp"
#include "select_device.hpp"
 
// HIP include(s).
#include <hip/hip_runtime_api.h>
 
namespace vecmem::hip::details {
 
class run_on_device {
 
public:
    run_on_device(int device) : m_device(device) {}
 
    template <typename EXECUTABLE>
    void operator()(EXECUTABLE exe) {
 
        // Switch to the device in a RAII mode.
        select_device helper(m_device);
 
        // Execute the code.
        exe();
    }
 
private:
    const int m_device;
 
};  // class run_on_device
 
}  // namespace vecmem::hip::details