From 0999ce9f0412874d155691ec781c8794e35461fe Mon Sep 17 00:00:00 2001
From: bingoo <1575938147@qq.com>
Date: Mon, 18 May 2026 12:07:01 +0800
Subject: [PATCH] adaptation triton

---
 flashinfer/__init__.py | 5 +++++
 requirements.txt       | 1 +
 2 files changed, 6 insertions(+)

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