From 7f112caac2ff365c3d6e0020fefe8c1300311e07 Mon Sep 17 00:00:00 2001
From: Younes Belkada <49240599+younesbelkada@users.noreply.github.com>
Date: Tue, 10 Sep 2024 11:14:54 +0400
Subject: [PATCH] Fix import of `FalconMambaForCausalLM` (#33381)

* fix build issues with FM kernels

* try another approach

* test

* fix

* add init files

* push fix

* fix

* fixup

* fix duplicate

* fix

* fix

* fix
---
 src/transformers/kernels/__init__.py | 0
 utils/check_build.py                 | 3 +++
 utils/check_inits.py                 | 1 +
 3 files changed, 4 insertions(+)
 create mode 100644 src/transformers/kernels/__init__.py

diff --git a/src/transformers/kernels/__init__.py b/src/transformers/kernels/__init__.py
new file mode 100644
index 000000000..e69de29bb
diff --git a/utils/check_build.py b/utils/check_build.py
index e3cca31f8..9ac309bd6 100644
--- a/utils/check_build.py
+++ b/utils/check_build.py
@@ -23,6 +23,9 @@ FILES_TO_FIND = [
     "kernels/rwkv/wkv_op.cpp",
     "kernels/deformable_detr/ms_deform_attn.h",
     "kernels/deformable_detr/cuda/ms_deform_im2col_cuda.cuh",
+    "kernels/falcon_mamba/selective_scan_with_ln_interface.py",
+    "kernels/falcon_mamba/__init__.py",
+    "kernels/__init__.py",
     "models/graphormer/algos_graphormer.pyx",
 ]
 
diff --git a/utils/check_inits.py b/utils/check_inits.py
index a9a841f3b..840bad086 100644
--- a/utils/check_inits.py
+++ b/utils/check_inits.py
@@ -333,6 +333,7 @@ IGNORE_SUBMODULES = [
     "safetensors_conversion",
     "modeling_gguf_pytorch_utils",
     "kernels.falcon_mamba",
+    "kernels",
 ]