mirror of
https://gitlab.kitware.com/vtk/vtk-m
synced 2024-10-05 18:08:59 +00:00
62c5095209
Previously, when you got a host/control portal from `ArrayHandleVirtual`, you got a version of an `ArrayPortal` that manged its own reference to the virtual structure that was implementing the portal. This was not done for device/execution portals because those objects could not do the appropriate resource management from the execution environment. Rather than releasing the host object to the portal, keep the host object managed by `StorageVirtual`. This allows the control and execution portals to be the same type, which we will need to be friendly with new array objects.
65 lines
2.0 KiB
C++
65 lines
2.0 KiB
C++
//============================================================================
|
|
// Copyright (c) Kitware, Inc.
|
|
// All rights reserved.
|
|
// See LICENSE.txt for details.
|
|
//
|
|
// This software is distributed WITHOUT ANY WARRANTY; without even
|
|
// the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
|
|
// PURPOSE. See the above copyright notice for more information.
|
|
//============================================================================
|
|
#ifndef vtk_m_cont_internal_TransferInfo_h
|
|
#define vtk_m_cont_internal_TransferInfo_h
|
|
|
|
#include <vtkm/cont/vtkm_cont_export.h>
|
|
|
|
#include <vtkm/Types.h>
|
|
#include <vtkm/cont/DeviceAdapterTag.h>
|
|
#include <vtkm/internal/ArrayPortalVirtual.h>
|
|
|
|
#include <memory>
|
|
|
|
namespace vtkm
|
|
{
|
|
|
|
namespace internal
|
|
{
|
|
class PortalVirtualBase;
|
|
}
|
|
|
|
namespace cont
|
|
{
|
|
namespace internal
|
|
{
|
|
|
|
struct VTKM_CONT_EXPORT TransferInfoArray
|
|
{
|
|
bool valid(vtkm::cont::DeviceAdapterId tagValue) const noexcept;
|
|
|
|
void updateHost(std::unique_ptr<vtkm::internal::PortalVirtualBase>&& host) noexcept;
|
|
void updateDevice(
|
|
vtkm::cont::DeviceAdapterId id,
|
|
std::unique_ptr<vtkm::internal::PortalVirtualBase>&& host_copy, //NOT the same as host version
|
|
const vtkm::internal::PortalVirtualBase* device,
|
|
const std::shared_ptr<void>& state) noexcept;
|
|
void releaseDevice();
|
|
void releaseAll();
|
|
|
|
const vtkm::internal::PortalVirtualBase* hostPtr() noexcept { return this->Host.get(); }
|
|
const vtkm::internal::PortalVirtualBase* devicePtr() const noexcept { return this->Device; }
|
|
vtkm::cont::DeviceAdapterId deviceId() const noexcept { return this->DeviceId; }
|
|
|
|
std::shared_ptr<void>& state() noexcept { return this->DeviceTransferState; }
|
|
|
|
private:
|
|
vtkm::cont::DeviceAdapterId DeviceId = vtkm::cont::DeviceAdapterTagUndefined{};
|
|
std::unique_ptr<vtkm::internal::PortalVirtualBase> Host = nullptr;
|
|
std::unique_ptr<vtkm::internal::PortalVirtualBase> HostCopyOfDevice = nullptr;
|
|
const vtkm::internal::PortalVirtualBase* Device = nullptr;
|
|
std::shared_ptr<void> DeviceTransferState = nullptr;
|
|
};
|
|
}
|
|
}
|
|
}
|
|
|
|
#endif
|