ThrustUtils.hpp

Go to the documentation of this file.

// This code is based on Jet framework.
// Copyright (c) 2018 Doyub Kim
// CubbyFlow is voxel-based fluid simulation engine for computer games.
// Copyright (c) 2020 CubbyFlow Team
// Core Part: Chris Ohk, Junwoo Hwang, Jihong Sin, Seungwoo Yoo
// AI Part: Dongheon Cho, Minseo Kim
// We are making my contributions/submissions to this project solely in our
// personal capacity and are not conveying any rights to any intellectual
// property of any third parties.

#ifndef CUBBYFLOW_THRUST_UTILS_HPP
#define CUBBYFLOW_THRUST_UTILS_HPP

#ifdef CUBBYFLOW_USE_CUDA

#include <Core/CUDA/CUDAArray.hpp>

#include <cuda_runtime.h>

#include <thrust/device_ptr.h>

#include <algorithm>
#include <cstdint>

namespace CubbyFlow
{
template <typename T, size_t N, typename D>
thrust::device_ptr<T> thrustBegin(CUDAArrayBase<T, N, D>& arr)
{
    return thrust::device_ptr<T>(arr.data());
}

template <typename T, size_t N, typename D>
thrust::device_ptr<const T> thrustCBegin(const CUDAArrayBase<T, N, D>& arr)
{
    return thrust::device_ptr<const T>(arr.data());
}

template <typename T, size_t N, typename D>
thrust::device_ptr<T> thrustEnd(CUDAArrayBase<T, N, D>& arr)
{
    return thrust::device_ptr<T>(arr.data() + arr.Length());
}

template <typename T, size_t N, typename D>
thrust::device_ptr<const T> thrustCEnd(const CUDAArrayBase<T, N, D>& arr)
{
    return thrust::device_ptr<const T>(arr.data() + arr.Length());
}
}  // namespace CubbyFlow

#endif

#endif