File: memory_info_impl.h

package info (click to toggle)
intel-compute-runtime 20.44.18297-1
  • links: PTS, VCS
  • area: main
  • in suites: bullseye
  • size: 34,780 kB
  • sloc: cpp: 379,729; lisp: 4,931; python: 299; sh: 196; makefile: 8
file content (51 lines) | stat: -rw-r--r-- 1,378 bytes parent folder | download
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
/*
 * Copyright (C) 2020 Intel Corporation
 *
 * SPDX-License-Identifier: MIT
 *
 */

#pragma once
#include "shared/source/helpers/basic_math.h"
#include "shared/source/helpers/debug_helpers.h"
#include "shared/source/os_interface/linux/drm_neo.h"
#include "shared/source/os_interface/linux/memory_info.h"

#include "drm/i915_drm.h"

#include <cstddef>
#include <cstdint>
#include <vector>

namespace NEO {

struct MemoryInfoImpl : public MemoryInfo {
    ~MemoryInfoImpl() override = default;

    MemoryInfoImpl(const drm_i915_memory_region_info *regionInfo, size_t count) : regions(regionInfo, regionInfo + count) {
    }

    drm_i915_gem_memory_class_instance getMemoryRegionClassAndInstance(uint32_t memoryBank) {
        auto index = (memoryBank > 0) ? Math::log2(memoryBank) + 1 : 0;
        if (index < regions.size()) {
            return regions[index].region;
        }
        return {invalidMemoryRegion(), invalidMemoryRegion()};
    }

    size_t getMemoryRegionSize(uint32_t memoryBank) {
        auto index = (memoryBank > 0) ? Math::log2(memoryBank) + 1 : 0;
        if (index < regions.size()) {
            return regions[index].probed_size;
        }
        return 0;
    }

    static constexpr uint16_t invalidMemoryRegion() {
        return static_cast<uint16_t>(-1);
    }

    std::vector<drm_i915_memory_region_info> regions;
};

} // namespace NEO