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
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
|
/* ************************************************************************
* Copyright (C) 2023 Advanced Micro Devices, Inc.
* ************************************************************************ */
#include <cstdlib>
#include <cstring>
#include <string>
#include "hipsolver.h"
#include "../rocblascommon/clients_utility.hpp"
#ifdef _WIN32
#define WIN32_LEAN_AND_MEAN
#include <windows.h>
#if __has_include(<filesystem>)
#include <filesystem>
namespace fs = std::filesystem;
#else
#include <experimental/filesystem>
namespace fs = std::experimental::filesystem;
#endif
#else
#include <fcntl.h>
#endif
// Return the path to the currently running executable
std::string hipsolver_exepath()
{
#ifdef _WIN32
std::vector<TCHAR> result(MAX_PATH + 1);
DWORD length = 0;
for(;;)
{
length = GetModuleFileNameA(nullptr, result.data(), result.size());
if(length < result.size() - 1)
{
result.resize(length + 1);
break;
}
result.resize(result.size() * 2);
}
fs::path exepath(result.begin(), result.end());
exepath = exepath.remove_filename();
exepath += exepath.empty() ? "" : "/";
return exepath.string();
#else
std::string pathstr;
char* path = realpath("/proc/self/exe", 0);
if(path)
{
char* p = strrchr(path, '/');
if(p)
{
p[1] = 0;
pathstr = path;
}
free(path);
}
return pathstr;
#endif
}
/* ============================================================================================ */
/* device query and print out their ID and name; return number of compute-capable devices. */
int query_device_property()
{
int device_count;
hipsolverStatus_t count_status = (hipsolverStatus_t)hipGetDeviceCount(&device_count);
if(count_status != HIPSOLVER_STATUS_SUCCESS)
{
printf("Query device error: cannot get device count \n");
return -1;
}
else
{
printf("Query device success: there are %d devices \n", device_count);
}
for(int i = 0; i < device_count; i++)
{
hipDeviceProp_t props;
hipsolverStatus_t props_status = (hipsolverStatus_t)hipGetDeviceProperties(&props, i);
if(props_status != HIPSOLVER_STATUS_SUCCESS)
{
printf("Query device error: cannot get device ID %d's property\n", i);
}
else
{
printf("Device ID %d : %s ------------------------------------------------------\n",
i,
props.name);
printf("with %3.1f GB memory, clock rate %dMHz @ computing capability %d.%d \n",
props.totalGlobalMem / 1e9,
(int)(props.clockRate / 1000),
props.major,
props.minor);
printf(
"maxGridDimX %d, sharedMemPerBlock %3.1f KB, maxThreadsPerBlock %d, warpSize %d\n",
props.maxGridSize[0],
props.sharedMemPerBlock / 1e3,
props.maxThreadsPerBlock,
props.warpSize);
printf("-------------------------------------------------------------------------\n");
}
}
return device_count;
}
/* set current device to device_id */
void set_device(int device_id)
{
hipsolverStatus_t status = (hipsolverStatus_t)hipSetDevice(device_id);
if(status != HIPSOLVER_STATUS_SUCCESS)
{
printf("Set device error: cannot set device ID %d, there may not be such device ID\n",
(int)device_id);
}
}
|
