memory_manager.hpp 1.82 KB
Newer Older
Nagy Mostafa's avatar
Nagy Mostafa committed
1
//*****************************************************************************
nmostafa's avatar
nmostafa committed
2
// Copyright 2017-2019 Intel Corporation
Nagy Mostafa's avatar
Nagy Mostafa committed
3 4 5 6 7 8 9 10 11 12 13 14 15 16
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//     http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
//*****************************************************************************

17 18 19
// NOTE: This file follows nGraph format style and MLIR naming convention since it does
// not expose public API to the rest of nGraph codebase and heavily depends on MLIR API.

Nagy Mostafa's avatar
Nagy Mostafa committed
20
#pragma once
21

Nagy Mostafa's avatar
Nagy Mostafa committed
22
#include <stdint.h>
23
#include <stdlib.h>
nmostafa's avatar
nmostafa committed
24
#include <vector>
25

Nagy Mostafa's avatar
Nagy Mostafa committed
26 27 28 29
namespace ngraph
{
    namespace runtime
    {
30
        namespace ngmlir
Nagy Mostafa's avatar
Nagy Mostafa committed
31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51
        {
            /// Memory manager for temporaries in MLIR compiled sub-graph
            /// It handles call-backs from the code and returns pointer to allocated memory
            /// Also, handles freeing up memory
            class MLIRMemMgr
            {
            public:
                /// Allocates data for temporary tensor. Currently, it is called for each
                /// temp tensor defintion. Keeps track of each pointer and free them during cleanup.
                // TODO: Use pre-allocation from framework memory manager
                void* allocate(size_t size);

                /// Frees all allocated pointers
                void freeAll();

            private:
                std::vector<void*> ptrList;
            };
        }
    }
}