|
1 | 1 | /*
|
2 | 2 | * Copyright (c) 2022 Advanced Micro Devices, Inc. All Rights reserved.
|
3 |
| - * Copyright (c) 2023 Triad National Security, LLC. All rights reserved. |
| 3 | + * Copyright (c) 2023-2024 Triad National Security, LLC. All rights reserved. |
4 | 4 | * $COPYRIGHT$
|
5 | 5 | *
|
6 | 6 | * Additional copyrights may follow
|
@@ -370,6 +370,32 @@ static int mca_accelerator_ze_record_event(int dev_id, opal_accelerator_event_t
|
370 | 370 | return OPAL_SUCCESS;
|
371 | 371 | }
|
372 | 372 |
|
| 373 | +static int mca_accelerator_ze_wait_event(int __opal_attribute_unused__ dev_id, |
| 374 | + opal_accelerator_event_t *event, |
| 375 | + opal_accelerator_stream_t * __opal_attribute_unused__ stream) |
| 376 | +{ |
| 377 | + ze_result_t zret; |
| 378 | + |
| 379 | + zret = zeEventHostSynchronize(*((ze_event_handle_t *)event->event), |
| 380 | + UINT64_MAX); |
| 381 | + switch (zret) { |
| 382 | + case ZE_RESULT_SUCCESS: |
| 383 | + return OPAL_SUCCESS; |
| 384 | + break; |
| 385 | + case ZE_RESULT_NOT_READY: |
| 386 | + return OPAL_ERR_RESOURCE_BUSY; |
| 387 | + break; |
| 388 | + default: |
| 389 | + opal_output_verbose(10, opal_accelerator_base_framework.framework_output, |
| 390 | + "zeEventHostSynchronize returned %d", zret); |
| 391 | + return OPAL_ERROR; |
| 392 | + } |
| 393 | + |
| 394 | + return OPAL_SUCCESS; |
| 395 | +} |
| 396 | + |
| 397 | + |
| 398 | + |
373 | 399 | static int mca_accelerator_ze_query_event(int dev_id, opal_accelerator_event_t *event)
|
374 | 400 | {
|
375 | 401 | ze_result_t zret;
|
@@ -411,7 +437,7 @@ static int mca_accelerator_ze_memcpy_async(int dest_dev_id, int src_dev_id, void
|
411 | 437 | }
|
412 | 438 |
|
413 | 439 | ze_stream = (opal_accelerator_ze_stream_t *)stream->stream;
|
414 |
| - assert(NULL != ze_stream); |
| 440 | + assert(NULL != ze_stream); |
415 | 441 |
|
416 | 442 | zret = zeCommandListAppendMemoryCopy(ze_stream->hCommandList,
|
417 | 443 | dest,
|
|
0 commit comments