Skip to content

Enable bitsandbytes packaging for ROCm#1299

Merged
Titus-von-Koeller merged 27 commits intobitsandbytes-foundation:multi-backend-refactorfrom
ROCm:device_abstraction
Aug 2, 2024
Merged

Enable bitsandbytes packaging for ROCm#1299
Titus-von-Koeller merged 27 commits intobitsandbytes-foundation:multi-backend-refactorfrom
ROCm:device_abstraction

Conversation

@pnunna93
Copy link
Copy Markdown
Contributor

This PR enables packaging wheels for bitsandbytes on ROCm. It updates rocm compilation and wheels build jobs to compile on ROCm 6.1 for latest Instinct and Radeon GPUs.

There are also updates to documentation based on user feedback and steps to install bitsandbytes ROCm package once available.

cc: @Titus-von-Koeller @sunway513 @amathews-amd

@pnunna93 pnunna93 mentioned this pull request Jul 31, 2024
7 tasks
@Titus-von-Koeller
Copy link
Copy Markdown
Collaborator

This looks very good! From my perspective I don't see anything amiss. @matthewdouglas could you take a second look, just to be sure and then we merge?

cc @akx if you have a moment and are interested to take a look :) we always highly appreciate your opinion!

Copy link
Copy Markdown
Contributor

@akx akx left a comment

Choose a reason for hiding this comment

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

Beyond the small suggestion about string formatting, should the naming be unified? What's HIP c.f. ROCm? :)

(Oh, and either the commit history here should be cleaned, or this should be squash-merged :) )

Comment thread bitsandbytes/cextension.py Outdated
Co-authored-by: Aarni Koskela <akx@iki.fi>
@pnunna93
Copy link
Copy Markdown
Contributor Author

Beyond the small suggestion about string formatting, should the naming be unified? What's HIP c.f. ROCm? :)

(Oh, and either the commit history here should be cleaned, or this should be squash-merged :) )

Thanks for the suggestion! HIP is a wrapper around ROCm, it supports both ROCm and CUDA platforms. The major/minor versions are same for HIP and ROCm.

Comment thread docs/source/installation.mdx Outdated
Comment thread docs/source/installation.mdx Outdated
Co-authored-by: Titus <9048635+Titus-von-Koeller@users.noreply.github.com>
@Titus-von-Koeller Titus-von-Koeller merged commit 6d9b69b into bitsandbytes-foundation:multi-backend-refactor Aug 2, 2024
@Titus-von-Koeller
Copy link
Copy Markdown
Collaborator

Thanks @pnunna93 great work and thanks a lot @akx and @matthewdouglas for helping with the review 🙌🏻

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

Projects

None yet

Development

Successfully merging this pull request may close these issues.

4 participants