AlpinDale
|
9d7beaa5b9
chore: separate kv_scale into k_scale and v_scale
|
5 months ago |
AlpinDale
|
156f577f79
feat: switch from `PYBIND11_MODULE` to `TORCH_LIBRARY` (#569)
|
6 months ago |
AlpinDale
|
3bdeb3e116
fix: clang formatting for all kernels (#558)
|
6 months ago |
AlpinDale
|
b746fb5562
fix a few warnings on the cpu kernels
|
6 months ago |
AlpinDale
|
8b56dc4347
dict -> torch.Tensor for blocks_to_swap
|
6 months ago |
AlpinDale
|
21ce19b3ea
blocks_to_copy dict -> torch.Tensor
|
6 months ago |
AlpinDale
|
9d81716bfd
[v0.5.3] Release Candidate (#388)
|
9 months ago |