Arrays.hpp

Go to the documentation of this file. Or view the newest version in sPHENIX GitHub for file Arrays.hpp

// This file is part of the Acts project.
//
// Copyright (C) 2020 CERN for the benefit of the Acts project
//
// This Source Code Form is subject to the terms of the Mozilla Public
// License, v. 2.0. If a copy of the MPL was not distributed with this
// file, You can obtain one at http://mozilla.org/MPL/2.0/.

#pragma once

// CUDA plugin include(s).
#include "Acts/Plugins/Cuda/Utilities/StreamWrapper.hpp"

// System include(s).
#include <cstddef>
#include <memory>

namespace Acts {
namespace Cuda {

namespace Details {

class DeviceArrayDeleter {
 public:
  void operator()(void* ptr);

};  // class DeviceArrayDeleter

class HostArrayDeleter {
 public:
  void operator()(void* ptr);

};  // class HostArrayDeleter

}  // namespace Details

template <typename T>
using device_array = std::unique_ptr<T, Details::DeviceArrayDeleter>;

template <typename T>
device_array<T> make_device_array(std::size_t size);

template <typename T>
using host_array = std::unique_ptr<T, Details::HostArrayDeleter>;

template <typename T>
host_array<T> make_host_array(std::size_t size);

template <typename T>
void copyToDevice(device_array<T>& dev, const host_array<T>& host,
                  std::size_t arraySize);

template <typename T>
void copyToDevice(device_array<T>& dev, const host_array<T>& host,
                  std::size_t arraySize, const StreamWrapper& stream);

template <typename T>
void copyToHost(host_array<T>& host, const device_array<T>& dev,
                std::size_t arraySize);

template <typename T>
void copyToHost(host_array<T>& host, const device_array<T>& dev,
                std::size_t arraySize, const StreamWrapper& stream);

}  // namespace Cuda
}  // namespace Acts