Skip to content

Commit 4d964a4

Browse files
callumfarecallumfare
committed
Generate .def file with all UR entry points
1 parent b766009 commit 4d964a4

File tree

3 files changed

+224
-0
lines changed

3 files changed

+224
-0
lines changed

‎include/ur_api_funcs.def

Lines changed: 188 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,188 @@
1+
2+
_UR_API(urPlatformGet)
3+
_UR_API(urPlatformGetInfo)
4+
_UR_API(urPlatformGetNativeHandle)
5+
_UR_API(urPlatformCreateWithNativeHandle)
6+
_UR_API(urPlatformGetApiVersion)
7+
_UR_API(urPlatformGetBackendOption)
8+
_UR_API(urContextCreate)
9+
_UR_API(urContextRetain)
10+
_UR_API(urContextRelease)
11+
_UR_API(urContextGetInfo)
12+
_UR_API(urContextGetNativeHandle)
13+
_UR_API(urContextCreateWithNativeHandle)
14+
_UR_API(urContextSetExtendedDeleter)
15+
_UR_API(urEventGetInfo)
16+
_UR_API(urEventGetProfilingInfo)
17+
_UR_API(urEventWait)
18+
_UR_API(urEventRetain)
19+
_UR_API(urEventRelease)
20+
_UR_API(urEventGetNativeHandle)
21+
_UR_API(urEventCreateWithNativeHandle)
22+
_UR_API(urEventSetCallback)
23+
_UR_API(urProgramCreateWithIL)
24+
_UR_API(urProgramCreateWithBinary)
25+
_UR_API(urProgramBuild)
26+
_UR_API(urProgramCompile)
27+
_UR_API(urProgramLink)
28+
_UR_API(urProgramRetain)
29+
_UR_API(urProgramRelease)
30+
_UR_API(urProgramGetFunctionPointer)
31+
_UR_API(urProgramGetGlobalVariablePointer)
32+
_UR_API(urProgramGetInfo)
33+
_UR_API(urProgramGetBuildInfo)
34+
_UR_API(urProgramSetSpecializationConstants)
35+
_UR_API(urProgramGetNativeHandle)
36+
_UR_API(urProgramCreateWithNativeHandle)
37+
_UR_API(urProgramBuildExp)
38+
_UR_API(urProgramCompileExp)
39+
_UR_API(urProgramLinkExp)
40+
_UR_API(urKernelCreate)
41+
_UR_API(urKernelGetInfo)
42+
_UR_API(urKernelGetGroupInfo)
43+
_UR_API(urKernelGetSubGroupInfo)
44+
_UR_API(urKernelRetain)
45+
_UR_API(urKernelRelease)
46+
_UR_API(urKernelGetNativeHandle)
47+
_UR_API(urKernelCreateWithNativeHandle)
48+
_UR_API(urKernelGetSuggestedLocalWorkSize)
49+
_UR_API(urKernelSetArgValue)
50+
_UR_API(urKernelSetArgLocal)
51+
_UR_API(urKernelSetArgPointer)
52+
_UR_API(urKernelSetExecInfo)
53+
_UR_API(urKernelSetArgSampler)
54+
_UR_API(urKernelSetArgMemObj)
55+
_UR_API(urKernelSetSpecializationConstants)
56+
_UR_API(urKernelSuggestMaxCooperativeGroupCountExp)
57+
_UR_API(urQueueGetInfo)
58+
_UR_API(urQueueCreate)
59+
_UR_API(urQueueRetain)
60+
_UR_API(urQueueRelease)
61+
_UR_API(urQueueGetNativeHandle)
62+
_UR_API(urQueueCreateWithNativeHandle)
63+
_UR_API(urQueueFinish)
64+
_UR_API(urQueueFlush)
65+
_UR_API(urSamplerCreate)
66+
_UR_API(urSamplerRetain)
67+
_UR_API(urSamplerRelease)
68+
_UR_API(urSamplerGetInfo)
69+
_UR_API(urSamplerGetNativeHandle)
70+
_UR_API(urSamplerCreateWithNativeHandle)
71+
_UR_API(urMemImageCreate)
72+
_UR_API(urMemBufferCreate)
73+
_UR_API(urMemRetain)
74+
_UR_API(urMemRelease)
75+
_UR_API(urMemBufferPartition)
76+
_UR_API(urMemGetNativeHandle)
77+
_UR_API(urMemBufferCreateWithNativeHandle)
78+
_UR_API(urMemImageCreateWithNativeHandle)
79+
_UR_API(urMemGetInfo)
80+
_UR_API(urMemImageGetInfo)
81+
_UR_API(urPhysicalMemCreate)
82+
_UR_API(urPhysicalMemRetain)
83+
_UR_API(urPhysicalMemRelease)
84+
_UR_API(urAdapterGet)
85+
_UR_API(urAdapterRelease)
86+
_UR_API(urAdapterRetain)
87+
_UR_API(urAdapterGetLastError)
88+
_UR_API(urAdapterGetInfo)
89+
_UR_API(urEnqueueKernelLaunch)
90+
_UR_API(urEnqueueEventsWait)
91+
_UR_API(urEnqueueEventsWaitWithBarrier)
92+
_UR_API(urEnqueueMemBufferRead)
93+
_UR_API(urEnqueueMemBufferWrite)
94+
_UR_API(urEnqueueMemBufferReadRect)
95+
_UR_API(urEnqueueMemBufferWriteRect)
96+
_UR_API(urEnqueueMemBufferCopy)
97+
_UR_API(urEnqueueMemBufferCopyRect)
98+
_UR_API(urEnqueueMemBufferFill)
99+
_UR_API(urEnqueueMemImageRead)
100+
_UR_API(urEnqueueMemImageWrite)
101+
_UR_API(urEnqueueMemImageCopy)
102+
_UR_API(urEnqueueMemBufferMap)
103+
_UR_API(urEnqueueMemUnmap)
104+
_UR_API(urEnqueueUSMFill)
105+
_UR_API(urEnqueueUSMMemcpy)
106+
_UR_API(urEnqueueUSMPrefetch)
107+
_UR_API(urEnqueueUSMAdvise)
108+
_UR_API(urEnqueueUSMFill2D)
109+
_UR_API(urEnqueueUSMMemcpy2D)
110+
_UR_API(urEnqueueDeviceGlobalVariableWrite)
111+
_UR_API(urEnqueueDeviceGlobalVariableRead)
112+
_UR_API(urEnqueueReadHostPipe)
113+
_UR_API(urEnqueueWriteHostPipe)
114+
_UR_API(urEnqueueKernelLaunchCustomExp)
115+
_UR_API(urEnqueueCooperativeKernelLaunchExp)
116+
_UR_API(urEnqueueTimestampRecordingExp)
117+
_UR_API(urEnqueueNativeCommandExp)
118+
_UR_API(urBindlessImagesUnsampledImageHandleDestroyExp)
119+
_UR_API(urBindlessImagesSampledImageHandleDestroyExp)
120+
_UR_API(urBindlessImagesImageAllocateExp)
121+
_UR_API(urBindlessImagesImageFreeExp)
122+
_UR_API(urBindlessImagesUnsampledImageCreateExp)
123+
_UR_API(urBindlessImagesSampledImageCreateExp)
124+
_UR_API(urBindlessImagesImageCopyExp)
125+
_UR_API(urBindlessImagesImageGetInfoExp)
126+
_UR_API(urBindlessImagesMipmapGetLevelExp)
127+
_UR_API(urBindlessImagesMipmapFreeExp)
128+
_UR_API(urBindlessImagesImportExternalMemoryExp)
129+
_UR_API(urBindlessImagesMapExternalArrayExp)
130+
_UR_API(urBindlessImagesMapExternalLinearMemoryExp)
131+
_UR_API(urBindlessImagesReleaseExternalMemoryExp)
132+
_UR_API(urBindlessImagesImportExternalSemaphoreExp)
133+
_UR_API(urBindlessImagesReleaseExternalSemaphoreExp)
134+
_UR_API(urBindlessImagesWaitExternalSemaphoreExp)
135+
_UR_API(urBindlessImagesSignalExternalSemaphoreExp)
136+
_UR_API(urUSMHostAlloc)
137+
_UR_API(urUSMDeviceAlloc)
138+
_UR_API(urUSMSharedAlloc)
139+
_UR_API(urUSMFree)
140+
_UR_API(urUSMGetMemAllocInfo)
141+
_UR_API(urUSMPoolCreate)
142+
_UR_API(urUSMPoolRetain)
143+
_UR_API(urUSMPoolRelease)
144+
_UR_API(urUSMPoolGetInfo)
145+
_UR_API(urUSMPitchedAllocExp)
146+
_UR_API(urUSMImportExp)
147+
_UR_API(urUSMReleaseExp)
148+
_UR_API(urCommandBufferCreateExp)
149+
_UR_API(urCommandBufferRetainExp)
150+
_UR_API(urCommandBufferReleaseExp)
151+
_UR_API(urCommandBufferFinalizeExp)
152+
_UR_API(urCommandBufferAppendKernelLaunchExp)
153+
_UR_API(urCommandBufferAppendUSMMemcpyExp)
154+
_UR_API(urCommandBufferAppendUSMFillExp)
155+
_UR_API(urCommandBufferAppendMemBufferCopyExp)
156+
_UR_API(urCommandBufferAppendMemBufferWriteExp)
157+
_UR_API(urCommandBufferAppendMemBufferReadExp)
158+
_UR_API(urCommandBufferAppendMemBufferCopyRectExp)
159+
_UR_API(urCommandBufferAppendMemBufferWriteRectExp)
160+
_UR_API(urCommandBufferAppendMemBufferReadRectExp)
161+
_UR_API(urCommandBufferAppendMemBufferFillExp)
162+
_UR_API(urCommandBufferAppendUSMPrefetchExp)
163+
_UR_API(urCommandBufferAppendUSMAdviseExp)
164+
_UR_API(urCommandBufferEnqueueExp)
165+
_UR_API(urCommandBufferRetainCommandExp)
166+
_UR_API(urCommandBufferReleaseCommandExp)
167+
_UR_API(urCommandBufferUpdateKernelLaunchExp)
168+
_UR_API(urCommandBufferGetInfoExp)
169+
_UR_API(urCommandBufferCommandGetInfoExp)
170+
_UR_API(urUsmP2PEnablePeerAccessExp)
171+
_UR_API(urUsmP2PDisablePeerAccessExp)
172+
_UR_API(urUsmP2PPeerAccessGetInfoExp)
173+
_UR_API(urVirtualMemGranularityGetInfo)
174+
_UR_API(urVirtualMemReserve)
175+
_UR_API(urVirtualMemFree)
176+
_UR_API(urVirtualMemMap)
177+
_UR_API(urVirtualMemUnmap)
178+
_UR_API(urVirtualMemSetAccess)
179+
_UR_API(urVirtualMemGetInfo)
180+
_UR_API(urDeviceGet)
181+
_UR_API(urDeviceGetInfo)
182+
_UR_API(urDeviceRetain)
183+
_UR_API(urDeviceRelease)
184+
_UR_API(urDevicePartition)
185+
_UR_API(urDeviceSelectBinary)
186+
_UR_API(urDeviceGetNativeHandle)
187+
_UR_API(urDeviceCreateWithNativeHandle)
188+
_UR_API(urDeviceGetGlobalTimestamps)

‎scripts/generate_code.py

Lines changed: 21 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -108,6 +108,26 @@ def _mako_print_cpp(path, namespace, tags, version, specs, meta):
108108
specs=specs,
109109
meta=meta)
110110

111+
112+
def _mako_api_funcs(path, namespace, tags, version, revision, specs, meta):
113+
template = "api_funcs.def.mako"
114+
fin = os.path.join(templates_dir, template)
115+
116+
name = "%s_api_funcs"%(namespace)
117+
filename = "%s.def"%(name)
118+
fout = os.path.join(path, filename)
119+
120+
print("Generating %s..."%fout)
121+
return util.makoWrite(
122+
fin, fout,
123+
name=name,
124+
ver=version,
125+
rev=revision,
126+
namespace=namespace,
127+
tags=tags,
128+
specs=specs,
129+
meta=meta)
130+
111131
"""
112132
generates c/c++ files from the specification documents
113133
"""
@@ -116,6 +136,7 @@ def _generate_api_cpp(incpath, srcpath, namespace, tags, version, revision, spec
116136
loc += _mako_api_cpp(srcpath, namespace, tags, version, revision, specs, meta)
117137
loc += _mako_ddi_h(incpath, namespace, tags, version, revision, specs, meta)
118138
loc += _mako_print_hpp(incpath, namespace, tags, version, revision, specs, meta)
139+
loc += _mako_api_funcs(incpath, namespace, tags, version, revision, specs, meta)
119140

120141
return loc
121142

‎scripts/templates/api_funcs.def.mako

Lines changed: 15 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,15 @@
1+
<%!
2+
import re
3+
from templates import helper as th
4+
%><%
5+
n=namespace
6+
N=n.upper()
7+
8+
x=tags['$x']
9+
X=x.upper()
10+
%>
11+
%for tbl in th.get_pfntables(specs, meta, n, tags):
12+
%for obj in tbl['functions']:
13+
_UR_API(${th.make_func_name(n, tags, obj)})
14+
%endfor
15+
%endfor

0 commit comments

Comments
 (0)