fix

younesbelkada · Sep 9, 2024 · 113c6fd · 113c6fd
1 parent 95d63a9
commit 113c6fd
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/utils/check_build.py b/utils/check_build.py
@@ -23,8 +23,8 @@
     "kernels/rwkv/wkv_op.cpp",
     "kernels/deformable_detr/ms_deform_attn.h",
     "kernels/deformable_detr/cuda/ms_deform_im2col_cuda.cuh",
-    "kernels/falcon_mamba/cuda/selective_scan_with_ln.py",
-    "kernels/falcon_mamba/cuda/__init__.py",
+    "kernels/falcon_mamba/selective_scan_with_ln.py",
+    "kernels/falcon_mamba/__init__.py",
     "kernels/deformable_detr/cuda/ms_deform_im2col_cuda.cuh",
     "models/graphormer/algos_graphormer.pyx",
 ]