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

Supports Bitsandbytes development on HPU (#117) #1714

Merged
merged 1 commit into from
Feb 5, 2025

Conversation

rsshaik1
Copy link
Contributor

@rsshaik1 rsshaik1 commented Jan 24, 2025

This PR adds changes required to support usage of Bitsandbytes on Gaudi2/3. Note that feature is still experimental and Gaudi specific changes to Bitsandbytes repo have not been upstreamed yet, therefore unit-tests added will not work for now. We expect to get these working in near future.

@rsshaik1 rsshaik1 marked this pull request as ready for review January 24, 2025 04:25
@vivekgoe vivekgoe self-requested a review January 24, 2025 04:56
@vivekgoe
Copy link
Collaborator

@regisss @libinta please help review this.

Copy link
Collaborator

@vivekgoe vivekgoe left a comment

Choose a reason for hiding this comment

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

LGTM.

import torch


def gaudi_check_device_same(first_device, second_device):
Copy link
Contributor

Choose a reason for hiding this comment

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

Consider adding typehints to make it easier to understand.

@libinta libinta added the run-test Run CI for PRs from external contributors label Feb 5, 2025
Copy link
Collaborator

@regisss regisss left a comment

Choose a reason for hiding this comment

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

LGTM!

@regisss regisss merged commit d15132c into huggingface:main Feb 5, 2025
4 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
run-test Run CI for PRs from external contributors
Projects
None yet
Development

Successfully merging this pull request may close these issues.

5 participants