diff --git a/dsv4/ops/rope_cuda.py b/dsv4/ops/rope_cuda.py index 292b0f84..72f86e53 100644 --- 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