Code

Add ROCm support#756

Code

Closed

arlo-phoenix wants to merge 8 commits intobitsandbytes-foundation:mainbitsandbytes-foundation/bitsandbytes:mainfrom arlo-phoenix:rocmarlo-phoenix/bitsandbytes-rocm-5.6:rocmCopy head branch name to clipboard

+159-32

Commits

Commits on Jan 22, 2024

• Add ROCm make target

  Show description for 5ee2ec1
  

  3 peoplecommitted
  5ee2ec1

  Copy full SHA for 5ee2ec1

• Add ROCm support to python library

  Show description for f9e2a84
  arlo-phoenix

  committed
  f9e2a84

  Copy full SHA for f9e2a84

• Add ROCm information to README

  arlo-phoenix

  committed
  78eecb3

  Copy full SHA for 78eecb3

• Makefile: Add fallback to /opt/rocm home

  arlo-phoenix

  committed
  940c52e

  Copy full SHA for 940c52e

• make cut one less

  arlo-phoenix

  committed
  a485a02

  Copy full SHA for a485a02

• Rename BITS_AND_BYTES_USE_ROCM to BNB_USE_HIP

  arlo-phoenix

  committed
  32cd5e0

  Copy full SHA for 32cd5e0

• Adjust kQuantizeBlockwise to work with WARP size 64

  arlo-phoenix

  committed
  e03a8bd

  Copy full SHA for e03a8bd

• Make sure DATA_NUM_PER_TH <= CUB_NUM_PER_TH

  Show description for ac20c05
  arlo-phoenix

  committed
  ac20c05

  Copy full SHA for ac20c05