Mila
Deep Neural Network Library
|
#include <cuda_runtime.h>
#include <memory_resource>
#include <string>
#include <source_location>
#include <iostream>
#include <sstream>
#include <cassert>
import Cuda.Error;
import Cuda.BadAlloc;
import Compute.MemoryResourceProperties;
import Compute.MemoryResource;
Classes | |
class | Mila::Dnn::Compute::CudaMemoryResource |
A memory resource that allocates memory on a CUDA device. More... | |
Namespaces | |
namespace | Mila |
namespace | Mila::Dnn |
namespace | Mila::Dnn::Compute |
Typedefs | |
using | Mila::Dnn::Compute::DeviceMemoryResource = CudaMemoryResource |
Alias for CudaMemoryResource that represents device-accessible memory. | |