machine.cpp
1.64 KB
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
52
53
54
55
56
/*===--------------------------------------------------------------------------
* ATMI (Asynchronous Task and Memory Interface)
*
* This file is distributed under the MIT License. See LICENSE.txt for details.
*===------------------------------------------------------------------------*/
#include "machine.h"
#include "atmi_runtime.h"
#include "internal.h"
#include <cassert>
#include <hsa.h>
#include <hsa_ext_amd.h>
#include <stdio.h>
#include <stdlib.h>
#include <vector>
extern ATLMachine g_atl_machine;
extern hsa_region_t atl_cpu_kernarg_region;
void ATLProcessor::addMemory(const ATLMemory &mem) {
for (auto &mem_obj : memories_) {
// if the memory already exists, then just return
if (mem.memory().handle == mem_obj.memory().handle)
return;
}
memories_.push_back(mem);
}
const std::vector<ATLMemory> &ATLProcessor::memories() const {
return memories_;
}
template <> std::vector<ATLCPUProcessor> &ATLMachine::processors() {
return cpu_processors_;
}
template <> std::vector<ATLGPUProcessor> &ATLMachine::processors() {
return gpu_processors_;
}
hsa_amd_memory_pool_t get_memory_pool(const ATLProcessor &proc,
const int mem_id) {
hsa_amd_memory_pool_t pool;
const std::vector<ATLMemory> &mems = proc.memories();
assert(mems.size() && mem_id >= 0 && mem_id < mems.size() &&
"Invalid memory pools for this processor");
pool = mems[mem_id].memory();
return pool;
}
template <> void ATLMachine::addProcessor(const ATLCPUProcessor &p) {
cpu_processors_.push_back(p);
}
template <> void ATLMachine::addProcessor(const ATLGPUProcessor &p) {
gpu_processors_.push_back(p);
}