PR #756 Add ROCm support

Add ROCm support #756

arlo-phoenix wants to merge 8 commits into bitsandbytes-foundation:main from arlo-phoenix:rocm

arlo-phoenix force pushed from a13af454 to 667e2837 2 years ago

arlo-phoenix force pushed from 667e2837 to 403ec745 2 years ago

arlo-phoenix force pushed from 403ec745 to bbdeb949 2 years ago

arlo-phoenix force pushed from bbdeb949 to 97f28638 2 years ago

arlo-phoenix force pushed from 97f28638 to 07230fd9 2 years ago

arlo-phoenix force pushed from 7c77be42 to 3c9c262e 2 years ago

Titus-von-Koeller added ROCm

TimDettmers added High Priority

TimDettmers added High Risk

TimDettmers assigned

younesbelkada 1 year ago

TimDettmers assigned

Titus-von-Koeller 1 year ago

Add ROCm make target

5ee2ec1b

Add ROCm support to python library

f9e2a843

Add ROCm information to README

78eecb38

Makefile: Add fallback to /opt/rocm home

940c52ed

make cut one less

a485a026

Rename BITS_AND_BYTES_USE_ROCM to BNB_USE_HIP

32cd5e0a

Adjust kQuantizeBlockwise to work with WARP size 64

e03a8bdd

arlo-phoenix force pushed from 62fa8a8e to e03a8bdd 1 year ago

Make sure DATA_NUM_PER_TH <= CUB_NUM_PER_TH

ac20c059

arlo-phoenix closed this 1 year ago

Reviewers

No reviews

Assignees

Titus-von-Koeller

younesbelkada

Labels

High Priority ROCm High Risk

Milestone

No milestone