stream.hpp

// SPDX-FileCopyrightText: 2017 - 2024 The Ginkgo authors
//
// SPDX-License-Identifier: BSD-3-Clause
 
#ifndef GKO_PUBLIC_CORE_BASE_STREAM_HPP_
#define GKO_PUBLIC_CORE_BASE_STREAM_HPP_
 
 
#include <ginkgo/core/base/executor.hpp>
 
 
namespace gko {
 

class cuda_stream {
public:
    cuda_stream();

    cuda_stream(int device_id);

    ~cuda_stream();
 
    cuda_stream(const cuda_stream&) = delete;

    cuda_stream(cuda_stream&&);
 
    cuda_stream& operator=(const cuda_stream&) = delete;

    cuda_stream& operator=(cuda_stream&&) = delete;

    CUstream_st* get() const;
 
private:
    CUstream_st* stream_;
 
    int device_id_;
};
 

class hip_stream {
public:
    hip_stream();

    hip_stream(int device_id);

    ~hip_stream();
 
    hip_stream(const hip_stream&) = delete;

    hip_stream(hip_stream&&);
 
    hip_stream& operator=(const hip_stream&) = delete;

    hip_stream& operator=(hip_stream&&) = delete;

    GKO_HIP_STREAM_STRUCT* get() const;
 
private:
    GKO_HIP_STREAM_STRUCT* stream_;
 
    int device_id_;
};
 
 
}  // namespace gko
 
 
#endif  // GKO_PUBLIC_CORE_BASE_STREAM_HPP_