diff --git a/flashinfer/jit/cpp_ext.py b/flashinfer/jit/cpp_ext.py
index fb0c40c00e..ef92a3c212 100644
--- a/flashinfer/jit/cpp_ext.py
+++ b/flashinfer/jit/cpp_ext.py
@@ -166,8 +166,11 @@ def generate_ninja_build_for_op(
         # No module flags, use global flags
         cuda_cflags += global_flags
 
+    # /usr/lib/wsl/lib is for WSL2 users, their cuda.so is there
     ldflags = [
         "-shared",
+        "-L/usr/lib/wsl/lib",
+        "-L$cuda_home/lib",
         "-L$cuda_home/lib64",
         "-L$cuda_home/lib64/stubs",
         "-lcudart",