Revision 4ff67af6a02a7501549a604d93e57feb5b6a3d45 authored by William Darling on 15 April 2016, 11:16:11 UTC, committed by William Darling on 25 April 2016, 10:18:05 UTC
1 parent 3529cf1
CUDAPageLockedMemAllocator.h
#pragma once
#include "MemAllocator.h"
namespace Microsoft { namespace MSR { namespace CNTK {
#ifdef _WIN32
#ifdef MATH_EXPORTS
#define MATH_API __declspec(dllexport)
#else
#define MATH_API __declspec(dllimport)
#endif
#else // no DLLs on Linux
#define MATH_API
#endif
class MATH_API CUDAPageLockedMemAllocator : public MemAllocator
{
public:
CUDAPageLockedMemAllocator(int deviceID);
int GetDeviceId() const;
void* Malloc(size_t size) override;
void Free(void* p) override;
static void* Malloc(size_t size, int deviceId);
static void Free(void* p, int deviceId);
private:
int m_deviceID;
};
} } }
Computing file changes ...
