#include <rmm/cuda_device.hpp>#include <rmm/cuda_stream_view.hpp>#include <rmm/detail/error.hpp>#include <rmm/detail/export.hpp>#include <rmm/detail/runtime_async_alloc.hpp>#include <rmm/detail/thrust_namespace.h>#include <rmm/mr/device/cuda_async_view_memory_resource.hpp>#include <rmm/mr/device/device_memory_resource.hpp>#include <cuda/std/type_traits>#include <cuda_runtime_api.h>#include <cstddef>#include <optional>
类 | |
| class | rmm::mr::cuda_async_memory_resource |
device_memory_resource 派生类,使用 cudaMallocAsync/cudaFreeAsync 进行分配/解除分配。更多... | |