adaptation triton

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()

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