diff --git a/flashinfer/__init__.py b/flashinfer/__init__.py
index 77b2c3e8f8..46e8dbc8cf 100644
--- a/flashinfer/__init__.py
+++ b/flashinfer/__init__.py
@@ -13,6 +13,9 @@
 See the License for the specific language governing permissions and
 limitations under the License.
 """
+import paddle
+
+paddle.enable_compat()
 
 import contextlib
 import importlib.util
@@ -192,3 +195,5 @@
 from .xqa import xqa_mla as xqa_mla
 from . import mamba as mamba
 from .fi_trace import fi_trace as fi_trace
+
+paddle.disable_compat()
diff --git a/requirements.txt b/requirements.txt
index 2544588256..018a5dd591 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,3 +1,4 @@
+triton
 apache-tvm-ffi>=0.1.6,!=0.1.8,!=0.1.8.post0,<0.2
 click
 cuda-tile