device_ptr.hip.hpp

Go to the documentation of this file.

 
#ifndef PLSSVM_BACKENDS_HIP_DETAIL_DEVICE_PTR_HPP_
#define PLSSVM_BACKENDS_HIP_DETAIL_DEVICE_PTR_HPP_
#pragma once
 
#include "plssvm/backends/gpu_device_ptr.hpp"  // plssvm::detail::gpu_device_ptr
 
namespace plssvm::hip::detail {
 
template <typename T>
class device_ptr : public ::plssvm::detail::gpu_device_ptr<T, int> {
    using base_type = ::plssvm::detail::gpu_device_ptr<T, int>;
 
    using base_type::data_;
    using base_type::queue_;
    using base_type::size_;
 
  public:
    // Be able to use overloaded base class functions.
    using base_type::memset;
    using base_type::fill;
    using base_type::copy_to_device;
    using base_type::copy_to_host;
 
    using typename base_type::const_host_pointer_type;
    using typename base_type::device_pointer_type;
    using typename base_type::host_pointer_type;
    using typename base_type::queue_type;
    using typename base_type::size_type;
    using typename base_type::value_type;
 
    device_ptr() = default;
    explicit device_ptr(size_type size, queue_type device = 0);
 
    device_ptr(const device_ptr &) = delete;
    device_ptr(device_ptr &&other) noexcept = default;
 
    device_ptr &operator=(const device_ptr &) = delete;
    device_ptr &operator=(device_ptr &&other) noexcept = default;
 
    ~device_ptr() override;
 
    void memset(int pattern, size_type pos, size_type num_bytes) override;
    void fill(value_type value, size_type pos, size_type count) override;
    void copy_to_device(const_host_pointer_type data_to_copy, size_type pos, size_type count) override;
    void copy_to_host(host_pointer_type buffer, size_type pos, size_type count) const override;
};
 
extern template class device_ptr<float>;
extern template class device_ptr<double>;
 
}  // namespace plssvm::hip::detail
 
#endif  // PLSSVM_BACKENDS_HIP_DETAIL_DEVICE_PTR_HPP_