Commit 65eb0f09 authored by pruthvi's avatar pruthvi

added documentation for the cpu allocator class

parent 2a3ebb35
...@@ -55,6 +55,8 @@ namespace ngraph ...@@ -55,6 +55,8 @@ namespace ngraph
} }
} }
// This class will be instantiated in the CPUCallFrame with the allocator object which will be used for the
// device memory allocation
class ngraph::runtime::cpu::CPUAllocator class ngraph::runtime::cpu::CPUAllocator
{ {
public: public:
...@@ -69,7 +71,7 @@ private: ...@@ -69,7 +71,7 @@ private:
size_t m_alignment; size_t m_alignment;
}; };
// Abstarct class for the allocator // Abstarct class for the allocator, for allocating and deallocating device memory
class ngraph::runtime::Allocator class ngraph::runtime::Allocator
{ {
public: public:
...@@ -77,6 +79,8 @@ public: ...@@ -77,6 +79,8 @@ public:
virtual void cpu_free(void* ptr, void*) = 0; virtual void cpu_free(void* ptr, void*) = 0;
}; };
// SystemAllocator overides and implements "cpu_malloc" & "cpu_free" of Alloctaor interface class
// this class uses system library malloc and free for device memory allocation
class ngraph::runtime::SystemAllocator : public ngraph::runtime::Allocator class ngraph::runtime::SystemAllocator : public ngraph::runtime::Allocator
{ {
public: public:
...@@ -108,6 +112,8 @@ private: ...@@ -108,6 +112,8 @@ private:
size_t m_alignment; size_t m_alignment;
}; };
// FrameworkAllocator overides and implements "cpu_malloc" & "cpu_free" of Alloctaor interface class,
// this class uses framework provide allocators and deallocators for device memory allocation
class ngraph::runtime::FrameworkAllocator : public ngraph::runtime::Allocator class ngraph::runtime::FrameworkAllocator : public ngraph::runtime::Allocator
{ {
public: public:
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment