device.hpp

// SPDX-FileCopyrightText: 2017 - 2024 The Ginkgo authors
//
// SPDX-License-Identifier: BSD-3-Clause
 
#ifndef GKO_PUBLIC_CORE_BASE_DEVICE_HPP_
#define GKO_PUBLIC_CORE_BASE_DEVICE_HPP_
 
 
#include <array>
#include <cstdint>
#include <mutex>
#include <type_traits>
 
#include <ginkgo/config.hpp>
 
 
namespace gko {
 
 
class CudaExecutor;
 
class HipExecutor;
 
 
class nvidia_device {
    friend class CudaExecutor;
    friend class HipExecutor;
 
private:
    static std::mutex& get_mutex(int i);
 
    static int& get_num_execs(int i);
 
    static constexpr int max_devices = 64;
};
 
 
class amd_device {
    friend class HipExecutor;
 
private:
    static std::mutex& get_mutex(int i);
 
    static int& get_num_execs(int i);
 
    static constexpr int max_devices = 64;
};
 
 
}  // namespace gko
 
#endif  // GKO_PUBLIC_CORE_BASE_DEVICE_HPP_