ROCm / bitsandbytes

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

adding arch detect to rocm6.2_internal_testing #22

Closed Lzy17 closed 6 months ago

Lzy17 commented 6 months ago

adding arch detection for rocm6.2_internal_testing

skipping only if arch=gfx90a