Skip to content

Commit c712c86

Browse files
Seanst98hdelannpmiller
authored andcommitted
Initial device side implementation for the sycl_ext_oneapi_async_memory_alloc extension (#16900)
Implement the [sycl_ext_oneapi_async_memory_alloc](intel/llvm#14800) extension for asynchronous memory allocation and freeing in CUDA, for device allocated pools only. SYCL entrypoints which specify host or shared side pools, or pools created by pre-existing allocations will throw. co-authored-by: Sean Stirling <sean.stirling@codeplay.com> co-authored-by: Hugh Delaney <hugh.delaney@codeplay.com> --------- Co-authored-by: Hugh Delaney <hugh.delaney@codeplay.com> Co-authored-by: Nicolas Miller <nicolas.miller@codeplay.com>
1 parent c35c877 commit c712c86

31 files changed

+713
-441
lines changed

include/ur_api.h

Lines changed: 49 additions & 43 deletions
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

include/ur_api_funcs.def

Lines changed: 1 addition & 1 deletion
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

include/ur_ddi.h

Lines changed: 7 additions & 6 deletions
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

include/ur_print.h

Lines changed: 10 additions & 10 deletions
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

include/ur_print.hpp

Lines changed: 31 additions & 32 deletions
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

scripts/core/EXP-ASYNC-ALLOC.rst

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -70,9 +70,9 @@ Functions
7070
* ${x}EnqueueUSMFreeExp
7171
* ${x}USMPoolCreateExp
7272
* ${x}USMPoolDestroyExp
73-
* ${x}USMPoolSetThresholdExp
7473
* ${x}USMPoolGetDefaultDevicePoolExp
7574
* ${x}USMPoolGetInfoExp
75+
* ${x}USMPoolSetInfoExp
7676
* ${x}USMPoolSetDevicePoolExp
7777
* ${x}USMPoolGetDevicePoolExp
7878
* ${x}USMPoolTrimToExp
@@ -94,6 +94,9 @@ Changelog
9494
| | DEVICE_INFO_ASYNC_USM_ALLOCATIONS_SUPPORT_EXP for |
9595
| | better consistency with other UR enums |
9696
+----------+----------------------------------------------------------+
97+
| 1.3 | Remove USMPoolSetThresholdExp |
98+
| | Replace with USMPoolSetInfoExp |
99+
+----------+----------------------------------------------------------+
97100

98101
Support
99102
--------------------------------------------------------------------------------

0 commit comments

Comments
 (0)