bnellnm 9f69856356 [Kernel] register punica functions as torch ops (#7591)

2024-08-16 13:59:38 -07: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] register punica functions as torch ops (#7591)

2024-08-16 13:59:38 -07:00

bgmv_expand.py

[Kernel] register punica functions as torch ops (#7591)

2024-08-16 13:59:38 -07:00

bgmv_shrink.py

[Kernel] register punica functions as torch ops (#7591)

2024-08-16 13:59:38 -07:00

sgmv_expand_slice.py

[Kernel] register punica functions as torch ops (#7591)

2024-08-16 13:59:38 -07:00

sgmv_expand.py

[Kernel] register punica functions as torch ops (#7591)

2024-08-16 13:59:38 -07:00

sgmv_shrink.py

[Kernel] register punica functions as torch ops (#7591)

2024-08-16 13:59:38 -07:00

utils.py

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

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