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

// SYCL include(s).
#include <CL/sycl.hpp>

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

namespace Acts::Sycl {

namespace detail {

class DeviceArrayDeleter {
 public:
  DeviceArrayDeleter(cl::sycl::queue& queue) : m_queue(&queue) {}
  void operator()(void* ptr) {
    if (ptr != nullptr) {
      cl::sycl::free(ptr, *m_queue);
    }
  }

 private:
  cl::sycl::queue* m_queue;
};  // class DeviceArrayDeleter

}  // namespace detail

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

template <typename T>
device_array<T> make_device_array(std::size_t size, cl::sycl::queue& queue) {
  return device_array<T>(cl::sycl::malloc_device<T>(size, queue),
                         detail::DeviceArrayDeleter(queue));
}

}  // namespace Acts::Sycl