fix: rope_cuda path — kernels/cuda not ops/cuda

2026-06-02 09:06:36 +00:00
parent 2bbbead984
commit 54a9b6961b
1 changed files with 1 additions and 1 deletions
--- a/dsv4/ops/rope_cuda.py
+++ b/dsv4/ops/rope_cuda.py
@@ -16,7 +16,7 @@ def _compile_and_load():
    if _LIB is not None:
        return _LIB
    
-    cu_path = Path(__file__).parent / "cuda" / "rope_cuda.cu"
+    cu_path = Path(__file__).parent.parent / "kernels" / "cuda" / "rope_cuda.cu"
    assert cu_path.exists(), f"rope_cuda.cu not found at {cu_path}"
    
    # Compile to shared library