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
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
|
//===--- amdgpu/impl/rt.h ----------------------------------------- C++ -*-===//
//
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
// See https://llvm.org/LICENSE.txt for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
//
//===----------------------------------------------------------------------===//
#ifndef SRC_RUNTIME_INCLUDE_RT_H_
#define SRC_RUNTIME_INCLUDE_RT_H_
#include "atmi_runtime.h"
#include "hsa_api.h"
#include <string>
namespace core {
#define DEFAULT_MAX_QUEUE_SIZE 4096
#define DEFAULT_DEBUG_MODE 0
class Environment {
public:
Environment()
: max_queue_size_(DEFAULT_MAX_QUEUE_SIZE),
debug_mode_(DEFAULT_DEBUG_MODE) {
GetEnvAll();
}
void GetEnvAll();
int getMaxQueueSize() const { return max_queue_size_; }
int getDebugMode() const { return debug_mode_; }
private:
std::string GetEnv(const char *name) {
char *env = getenv(name);
std::string ret;
if (env) {
ret = env;
}
return ret;
}
int max_queue_size_;
int debug_mode_;
};
class Runtime final {
public:
static Runtime &getInstance() {
static Runtime instance;
return instance;
}
// modules
static hsa_status_t RegisterModuleFromMemory(
void *, size_t, hsa_agent_t agent,
hsa_status_t (*on_deserialized_data)(void *data, size_t size,
void *cb_state),
void *cb_state, std::vector<hsa_executable_t> &HSAExecutables);
// data
static hsa_status_t Memcpy(hsa_signal_t, void *, const void *, size_t);
static hsa_status_t Memfree(void *);
static hsa_status_t HostMalloc(void **ptr, size_t size);
static hsa_status_t DeviceMalloc(void **ptr, size_t size, int DeviceId);
int getMaxQueueSize() const { return env_.getMaxQueueSize(); }
int getDebugMode() const { return env_.getDebugMode(); }
private:
static hsa_status_t Malloc(void **ptr, size_t size, int DeviceId,
atmi_devtype_t DeviceType);
protected:
Runtime() = default;
~Runtime() = default;
Runtime(const Runtime &) = delete;
Runtime &operator=(const Runtime &) = delete;
protected:
// variable to track environment variables
Environment env_;
};
} // namespace core
#endif // SRC_RUNTIME_INCLUDE_RT_H_
|
