kvcache-ai/ktransformers
1
0
Fork 0
mirror of https://github.com/kvcache-ai/ktransformers.git synced 2026-05-05 05:31:36 +00:00
Code Issues Packages Projects Releases Wiki Activity

[fix](test): fix import kt-kernel (#1728)

Browse Source
This commit is contained in:
ErvinXie
2025-12-17 19:46:32 +08:00
committed by GitHub
parent 6fc4080a7d
commit a8667ddb58
33 changed files with 1063 additions and 1151 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
kt-kernel/bench/bench_moe_kernel_tiling.py
View File

@@ -26,7 +26,7 @@ os.environ.setdefault("BLAS_NUM_THREADS", "1")
sys.path.insert(0, os.path.join(os.path.dirname(__file__), "..", "build"))
import torch # noqa: E402
import kt_kernel_ext as ce # noqa: E402
from kt_kernel import kt_kernel_ext as ce # noqa: E402
from tqdm import tqdm # noqa: E402