fix: correct gather.py kernel_dir path

2026-05-30 21:12:09 +00:00
parent 924707a673
commit 8fb9d89658
1 changed files with 1 additions and 1 deletions
--- a/dsv4/kernels/cache/gather.py
+++ b/dsv4/kernels/cache/gather.py
@@ -15,7 +15,7 @@ import torch
 from torch.utils.cpp_extension import load

 _kernel_module = None
-_kernel_dir = os.path.join(os.path.dirname(__file__), "cuda")
+_kernel_dir = os.path.join(os.path.dirname(__file__), "..", "cuda")


 def _get_kernel_module():