kvcache-ai-ktransformers/ktransformers/operators
Azure 3986e2d2cf
Merge pull request #178 from fxzjshm/hip
[Feat] Port to ROCm/HIP
2025-03-15 02:31:07 +08:00
..
__init__.py Initial commit 2024-07-27 16:06:58 +08:00
attention.py merge main; Add torch q8 linear 2025-03-14 05:52:07 -04:00
base_operator.py fix precision bug imported by position_ids in 0.2.0 2025-02-17 09:23:14 +00:00
cpuinfer.py cpuinfer: filter repeated backend instantiation 2025-03-10 22:03:04 +08:00
dynamic_attention.py merge main; Add torch q8 linear 2025-03-14 05:52:07 -04:00
experts.py fix-singleton 2025-03-14 04:16:53 +00:00
flashinfer_wrapper.py fix flashinfer precision 2025-03-07 14:07:00 +00:00
gate.py use compile for gate, slight performance improvement 2025-03-14 12:43:28 +00:00
linear.py Merge pull request #178 from fxzjshm/hip 2025-03-15 02:31:07 +08:00
models.py merge main; Add torch q8 linear 2025-03-14 05:52:07 -04:00
RoPE.py fix precision bug imported by position_ids in 0.2.0 2025-02-17 09:23:14 +00:00
triton_attention.py merge main; Add torch q8 linear 2025-03-14 05:52:07 -04:00
triton_attention_prefill.py merge main; Add torch q8 linear 2025-03-14 05:52:07 -04:00