Skip to content

Enable bitsandbytes packaging for ROCm - #1299

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

Enable bitsandbytes packaging for ROCm#1299
Titus-von-Koeller merged 27 commits into
bitsandbytes-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
Member

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!

@akx akx left a comment

Copy link
Copy Markdown
Contributor

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
Member

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