vrr/kvcache-ai-ktransformers
2
0
Fork 0
mirror of https://github.com/kvcache-ai/ktransformers.git synced 2026-04-28 03:39:48 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 4

[fix]: add amx optimization

Browse source
This commit is contained in:
KMSorSMS 2025-11-07 15:51:56 +00:00
parent 9509220b64
commit 2641d15383
3 changed files with 33 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

2
kt-kernel/ext_bindings.cpp
View file

@ -177,7 +177,7 @@ class MOEBindings {
const uintptr_t physical_to_logical_map = 0) {
Args* args = new Args{nullptr, moe.get()};
if (physical_to_logical_map) {
printf("debug physical_to_logical_map in arg:%lu\n", physical_to_logical_map);
// printf("debug physical_to_logical_map in arg:%lu\n", physical_to_logical_map);
moe->config.physical_to_logical_map = reinterpret_cast<void*>(physical_to_logical_map);
printf("moe ptr:%p,confirm: moe->config.physical_to_logical_map:%lu\n", reinterpret_cast<void*>(moe.get()),
reinterpret_cast<uintptr_t>(moe->config.physical_to_logical_map));