https://github.com/Microsoft/CNTK
Raw File
Tip revision: c6fc20ca4bffcce7bdbbf8a9a68581d89f1770ee authored by PengchengHe on 12 February 2017, 12:45:42 UTC
Implement cosdistance/cosdistance_drive in native CUDA code to optimize
Tip revision: c6fc20c
MemAllocator.h
#pragma once

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 MemAllocator
{
public:
    virtual void* Malloc(size_t size) = 0;
    virtual void Free(void* p) = 0;
};
} } }
back to top