Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[cuda] Move to hal/drivers and wire up BUILD files #14620

Merged
merged 1 commit into from
Jan 10, 2024

Conversation

antiagainst
Copy link
Contributor

@antiagainst antiagainst commented Aug 10, 2023

This commit moves the CUDA HAL driver rewrite to the
hal/drivers directory given it's functional and ready for
normal usage. By this we can start run tests with CI
to make sure it does not regress. Further improvements
can happen directly in this directory.

This provides an easy route for trying out the rewrite before
eventually replace the existing HAL driver.

Along the way wired up BUILD configurations.

Progress towards #13245

@antiagainst antiagainst added the hal/cuda Runtime CUDA HAL backend label Aug 10, 2023
@antiagainst antiagainst force-pushed the cuda2-to-hal-drivers branch 2 times, most recently from ff8ae89 to fadf0c8 Compare August 12, 2023 04:52
@antiagainst antiagainst marked this pull request as ready for review August 13, 2023 01:36
CMakeLists.txt Outdated Show resolved Hide resolved
runtime/src/iree/hal/drivers/cuda2/README.md Outdated Show resolved Hide resolved
Comment on lines 247 to +263
option(IREE_HAL_DRIVER_CUDA "Enables the 'cuda' runtime HAL driver" ${IREE_HAL_DRIVER_CUDA_DEFAULT})
option(IREE_HAL_DRIVER_CUDA2 "Enables the 'cuda2' runtime HAL driver" ${IREE_HAL_DRIVER_CUDA2_DEFAULT})
Copy link
Member

Choose a reason for hiding this comment

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

Can we come up with a plan for switching from the old cuda to the new cuda2? Maybe post an RFC and an update to #13245, and could link to one or both of those here.

Some ideas:

Ideally we'd have a timeline in mind so we don't stay in this state of having two similar paths for long.

Copy link
Contributor Author

Choose a reason for hiding this comment

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

SG. I've commented on #13245 with #13245 (comment). I'll send out a mail later too.

Copy link
Contributor Author

Choose a reason for hiding this comment

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

Hard for me to give a precise timeline though; given I'm multiplexing over quite a few work streams. But yeah agreed we don't want to be in a hybrid state for too long; I'm hoping to get feature parity down soon.

This commit moves the CUDA HAL driver rewrite to the
`hal/drivers` directory given it's functional and ready for
normal usage. By this we can start run tests with CI
to make sure it does not regress. Further improvements
can happen directly in this directory.

This provides an easy route for trying out the rewrite before
eventually replace the existing HAL driver.

Along the way wired up BUILD configurations.
@antiagainst
Copy link
Contributor Author

Hi @benvanik, I have addressed all known issues so this is ready for another review now. After moving I'd be switching the tests/benchmarks to use the new impl to test out.

@benvanik
Copy link
Collaborator

woo excited to get rid of the old impl!

@antiagainst antiagainst merged commit 171e31c into iree-org:main Jan 10, 2024
57 checks passed
@antiagainst antiagainst deleted the cuda2-to-hal-drivers branch January 10, 2024 00:55
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
hal/cuda Runtime CUDA HAL backend
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants