remove changes about import tvm_ffi

Author: SigureMo

flashinfer/fused_moe/core.py

@@ -20,10 +20,6 @@
 from typing import Any, Dict, List, Optional, Tuple, Union
 
 import torch
-import paddle
-
-with paddle.compat.use_torch_proxy_guard(enable=False):
-    import tvm_ffi
 
 from ..autotuner import (
     AutoTuner,
@@ -354,7 +350,9 @@ def __init__(
             )
             self.activation_type = activation_type
 
-            def paddle_dtype_to_tvm_ffi_dtype(dtype: paddle.dtype):
+            def paddle_dtype_to_tvm_ffi_dtype(dtype):
+                import tvm_ffi
+
                 dtype_str = str(dtype).split(".", 1)[-1]
                 return tvm_ffi.dtype(dtype_str)
 

flashinfer/jit/core.py

@@ -1,10 +1,6 @@
 import dataclasses
 import logging
 import os
-import paddle
-
-with paddle.compat.use_torch_proxy_guard(enable=False):
-    import tvm_ffi
 from contextlib import nullcontext
 from datetime import datetime
 from pathlib import Path

flashinfer/jit/cpp_ext.py

@@ -10,10 +10,7 @@
 from pathlib import Path
 from typing import List, Optional
 
-import paddle
-
-with paddle.compat.use_torch_proxy_guard(enable=False):
-    import tvm_ffi
+import tvm_ffi
 import torch
 
 from . import env as jit_env