Skip to content

Commit b183751

Browse files
Seanst98Hugh Delaney
andauthored
[AsyncAlloc][UR][Exp] Initial API for async alloc entry points (#17117)
Introduce unimplemented UR API for the creation and use of memory pools with enqueued allocs/frees. Currently, this only supports device allocated memory pools. This extension is experimental and may change drastically between revisions. --------- Co-authored-by: Hugh Delaney <hugh.delaney@codeplay.com>
1 parent cc0b20f commit b183751

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

43 files changed

+7728
-1206
lines changed

unified-runtime/include/ur_api.h

Lines changed: 1067 additions & 451 deletions
Large diffs are not rendered by default.

unified-runtime/include/ur_api_funcs.def

Lines changed: 24 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -132,9 +132,33 @@ _UR_API(urEnqueueReadHostPipe)
132132
_UR_API(urEnqueueWriteHostPipe)
133133
_UR_API(urEnqueueEventsWaitWithBarrierExt)
134134
_UR_API(urEnqueueKernelLaunchCustomExp)
135+
_UR_API(urEnqueueUSMDeviceAllocExp)
136+
_UR_API(urEnqueueUSMSharedAllocExp)
137+
_UR_API(urEnqueueUSMHostAllocExp)
138+
_UR_API(urEnqueueUSMFreeExp)
135139
_UR_API(urEnqueueCooperativeKernelLaunchExp)
136140
_UR_API(urEnqueueTimestampRecordingExp)
137141
_UR_API(urEnqueueNativeCommandExp)
142+
_UR_API(urUSMHostAlloc)
143+
_UR_API(urUSMDeviceAlloc)
144+
_UR_API(urUSMSharedAlloc)
145+
_UR_API(urUSMFree)
146+
_UR_API(urUSMGetMemAllocInfo)
147+
_UR_API(urUSMPoolCreate)
148+
_UR_API(urUSMPoolRetain)
149+
_UR_API(urUSMPoolRelease)
150+
_UR_API(urUSMPoolGetInfo)
151+
_UR_API(urUSMPoolCreateExp)
152+
_UR_API(urUSMPoolDestroyExp)
153+
_UR_API(urUSMPoolSetThresholdExp)
154+
_UR_API(urUSMPoolGetDefaultDevicePoolExp)
155+
_UR_API(urUSMPoolGetInfoExp)
156+
_UR_API(urUSMPoolSetDevicePoolExp)
157+
_UR_API(urUSMPoolGetDevicePoolExp)
158+
_UR_API(urUSMPoolTrimToExp)
159+
_UR_API(urUSMPitchedAllocExp)
160+
_UR_API(urUSMImportExp)
161+
_UR_API(urUSMReleaseExp)
138162
_UR_API(urBindlessImagesUnsampledImageHandleDestroyExp)
139163
_UR_API(urBindlessImagesSampledImageHandleDestroyExp)
140164
_UR_API(urBindlessImagesImageAllocateExp)
@@ -153,18 +177,6 @@ _UR_API(urBindlessImagesImportExternalSemaphoreExp)
153177
_UR_API(urBindlessImagesReleaseExternalSemaphoreExp)
154178
_UR_API(urBindlessImagesWaitExternalSemaphoreExp)
155179
_UR_API(urBindlessImagesSignalExternalSemaphoreExp)
156-
_UR_API(urUSMHostAlloc)
157-
_UR_API(urUSMDeviceAlloc)
158-
_UR_API(urUSMSharedAlloc)
159-
_UR_API(urUSMFree)
160-
_UR_API(urUSMGetMemAllocInfo)
161-
_UR_API(urUSMPoolCreate)
162-
_UR_API(urUSMPoolRetain)
163-
_UR_API(urUSMPoolRelease)
164-
_UR_API(urUSMPoolGetInfo)
165-
_UR_API(urUSMPitchedAllocExp)
166-
_UR_API(urUSMImportExp)
167-
_UR_API(urUSMReleaseExp)
168180
_UR_API(urCommandBufferCreateExp)
169181
_UR_API(urCommandBufferRetainExp)
170182
_UR_API(urCommandBufferReleaseExp)

unified-runtime/include/ur_ddi.h

Lines changed: 218 additions & 135 deletions
Large diffs are not rendered by default.

unified-runtime/include/ur_print.h

Lines changed: 260 additions & 120 deletions
Large diffs are not rendered by default.

0 commit comments

Comments
 (0)