1 //===--- amdgpu/impl/atmi_interop_hsa.cpp ------------------------- C++ -*-===//
2 //
3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4 // See https://llvm.org/LICENSE.txt for license information.
5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6 //
7 //===----------------------------------------------------------------------===//
8 #include "atmi_interop_hsa.h"
9 #include "internal.h"
10 #include "machine.h"
11
12 // TODO: need to get rid of this as well
13
14 extern ATLMachine g_atl_machine;
15
atmi_interop_hsa_get_symbol_info(const std::map<std::string,atl_symbol_info_t> & SymbolInfoTable,int DeviceId,const char * symbol,void ** var_addr,unsigned int * var_size)16 hsa_status_t atmi_interop_hsa_get_symbol_info(
17 const std::map<std::string, atl_symbol_info_t> &SymbolInfoTable,
18 int DeviceId, const char *symbol, void **var_addr, unsigned int *var_size) {
19 /*
20 // Typical usage:
21 void *var_addr;
22 size_t var_size;
23 atmi_interop_hsa_get_symbol_addr(gpu_place, "symbol_name", &var_addr,
24 &var_size);
25 atmi_memcpy(signal, host_add, var_addr, var_size);
26 */
27
28 if (!symbol || !var_addr || !var_size)
29 return HSA_STATUS_ERROR;
30
31 // get the symbol info
32 std::string symbolStr = std::string(symbol);
33 auto It = SymbolInfoTable.find(symbolStr);
34 if (It != SymbolInfoTable.end()) {
35 atl_symbol_info_t info = It->second;
36 *var_addr = reinterpret_cast<void *>(info.addr);
37 *var_size = info.size;
38 return HSA_STATUS_SUCCESS;
39 } else {
40 *var_addr = NULL;
41 *var_size = 0;
42 return HSA_STATUS_ERROR;
43 }
44 }
45
atmi_interop_hsa_get_kernel_info(const std::map<std::string,atl_kernel_info_t> & KernelInfoTable,int DeviceId,const char * kernel_name,hsa_executable_symbol_info_t kernel_info,uint32_t * value)46 hsa_status_t atmi_interop_hsa_get_kernel_info(
47 const std::map<std::string, atl_kernel_info_t> &KernelInfoTable,
48 int DeviceId, const char *kernel_name,
49 hsa_executable_symbol_info_t kernel_info, uint32_t *value) {
50 /*
51 // Typical usage:
52 uint32_t value;
53 atmi_interop_hsa_get_kernel_addr(gpu_place, "kernel_name",
54 HSA_EXECUTABLE_SYMBOL_INFO_KERNEL_KERNARG_SEGMENT_SIZE,
55 &val);
56 */
57
58 if (!kernel_name || !value)
59 return HSA_STATUS_ERROR;
60
61 hsa_status_t status = HSA_STATUS_SUCCESS;
62 // get the kernel info
63 std::string kernelStr = std::string(kernel_name);
64 auto It = KernelInfoTable.find(kernelStr);
65 if (It != KernelInfoTable.end()) {
66 atl_kernel_info_t info = It->second;
67 switch (kernel_info) {
68 case HSA_EXECUTABLE_SYMBOL_INFO_KERNEL_GROUP_SEGMENT_SIZE:
69 *value = info.group_segment_size;
70 break;
71 case HSA_EXECUTABLE_SYMBOL_INFO_KERNEL_PRIVATE_SEGMENT_SIZE:
72 *value = info.private_segment_size;
73 break;
74 case HSA_EXECUTABLE_SYMBOL_INFO_KERNEL_KERNARG_SEGMENT_SIZE:
75 // return the size for non-implicit args
76 *value = info.kernel_segment_size - sizeof(atmi_implicit_args_t);
77 break;
78 default:
79 *value = 0;
80 status = HSA_STATUS_ERROR;
81 break;
82 }
83 } else {
84 *value = 0;
85 status = HSA_STATUS_ERROR;
86 }
87
88 return status;
89 }
90