vrr/kvcache-ai-ktransformers
2
0
Fork 0
mirror of https://github.com/kvcache-ai/ktransformers.git synced 2025-09-09 22:05:30 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix build failure on msvc

Browse source
This commit is contained in:
akemimadoka 2025-04-30 21:51:49 +08:00
parent ee2ede0412
commit 206e51f77e
2 changed files with 8 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

5
setup.py
View file

@ -48,7 +48,10 @@ if dev_backend == "xpu":
"pytorch-triton-xpu==3.3.0"
]
else:
triton_dep = ["triton>=3.2"]
triton_dep = [
"triton >= 3.2; sys_platform != 'win32' and sys_platform != 'Windows'",
"triton-windows >= 3.2; sys_platform == 'win32' or sys_platform == 'Windows'"
]
with_balance = os.environ.get("USE_BALANCE_SERVE", "0") == "1"