Skip to content

Comments

[rocsolver] Use enqueue_native_command ext when avail#582

Merged
Rbiessy merged 1 commit intouxlfoundation:developfrom
JackAKirk:rocsolver-update-host-task
Oct 8, 2024
Merged

[rocsolver] Use enqueue_native_command ext when avail#582
Rbiessy merged 1 commit intouxlfoundation:developfrom
JackAKirk:rocsolver-update-host-task

Conversation

@JackAKirk
Copy link
Contributor

This makes use of the enqueue_native_command dpc++ extension if it is available. This improves performance and integrates correctly with the dpc++ scheduler.

The implementation is very similar to the cusolver part of #572.
See #572 for further details of this extension.

Since I made a small change to the standard host_task implementation to add a missing sync in the batch functions, I also attach tests (marked "*host_task") for the case that the extension macro for enqueue_native_command isn't defined.

tests:

test_main_lapack_rt_native_command_amd.txt
test_main_lapack_ct_native_command_amd.txt
test_main_lapack_ct_host_task_amd.txt
test_main_lapack_rt_host_task_amd.txt

This makes use of the enqueue_native_command dpc++ extension if it is available. This improves performance and integrates correctly with the dpc++ scheduler.

Signed-off-by: JackAKirk <jack.kirk@codeplay.com>
@JackAKirk
Copy link
Contributor Author

@oneapi-src/onemkl-lapack-write could you please review this?

Thanks

Copy link
Contributor

@sknepper sknepper left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

This looks great - thanks, @JackAKirk !

@Rbiessy Rbiessy merged commit 09d4ab3 into uxlfoundation:develop Oct 8, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants