ROCm / bitsandbytes

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

Enable igemmlt int test on rocm #3

Closed pnunna93 closed 10 months ago

pnunna93 commented 10 months ago

This PR has changes to enable igemmlt function and its dependencies.

pnunna93 commented 10 months ago

These are the test statistics before and after this PR:

Before igemmlt: Module Passed Failed Skipped
functional 261 34 73
autograd 680 24 1536
linear8bitlt 2 0 16
modules 10 3 1
optim 125 26 26
triton 0 0 2
Total 1078 87 1654
After igemmlt: Module Passed Failed Skipped
functional 286 23 27
autograd 1592 648 0
linear8bitlt 9 9 0
modules 10 4 0
optim 125 26 26
triton 0 0 2
Total 2022 710 55
Lzy17 commented 10 months ago

looks good to me!