yuehuayingxueluo 04aca9e55b [Inference/Kernel]Add get_cos_and_sin Kernel (#5528) ...

* Add get_cos_and_sin kernel

* fix code comments

* fix code typos

* merge common codes of get_cos_and_sin kernel.

* Fixed a typo

* Changed 'asset allclose' to 'assert equal'.

2024-04-01 13:47:14 +08:00

..

cuda

[Inference/Kernel]Add get_cos_and_sin Kernel (#5528)

2024-04-01 13:47:14 +08:00

triton

[Inference]Move benchmark-related code to the example directory. (#5408)

2024-02-28 16:46:03 +08:00

__init__.py

[Inference]Add CUDA KVCache Kernel (#5406)

2024-02-28 14:36:50 +08:00