ROCm / bitsandbytes

8-bit CUDA functions for PyTorch
MIT License
38 stars 4 forks source link

adding arch detection for test_gemv_eye_4bit #21

Closed Lzy17 closed 6 months ago

Lzy17 commented 6 months ago

adding arch detection for test_gemv_eye_4bit

skipping only if arch=gfx90a