Jee Jee Li 9b0e3ec970 [Kernel][LoRA] Add assertion for punica sgmv kernels (#7585)

2024-09-23 18:57:42 +00:00

..

__init__.py

[Kernel][RFC] Refactor the punica kernel based on Triton (#5036)

2024-07-31 17:12:24 -07:00

bgmv_expand_slice.py

[Kernel][LoRA] Add assertion for punica sgmv kernels (#7585)

2024-09-23 18:57:42 +00:00

bgmv_expand.py

[Kernel][LoRA] Add assertion for punica sgmv kernels (#7585)

2024-09-23 18:57:42 +00:00

bgmv_shrink.py

[Bugfix] Make torch registration of punica ops optional (#7970)

2024-08-28 16:11:49 -06:00

sgmv_expand_slice.py

[Kernel][LoRA] Add assertion for punica sgmv kernels (#7585)

2024-09-23 18:57:42 +00:00

sgmv_expand.py

[Kernel][LoRA] Add assertion for punica sgmv kernels (#7585)

2024-09-23 18:57:42 +00:00

sgmv_shrink.py

[Kernel][LoRA] Add assertion for punica sgmv kernels (#7585)

2024-09-23 18:57:42 +00:00

utils.py

[Kernel][RFC] Refactor the punica kernel based on Triton (#5036)

2024-07-31 17:12:24 -07:00