Skip to content

Commit 416c7d5

Browse files
committed
Revert "[Bug] Dynamo Unsupported due to BasevLLMParameter.torch_function ca…"
This reverts commit 4492e3a. Signed-off-by: yewentao256 <zhyanwentao@126.com>
1 parent 2e5df88 commit 416c7d5

File tree

1 file changed

+0
-10
lines changed

1 file changed

+0
-10
lines changed

vllm/model_executor/parameter.py

Lines changed: 0 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,6 @@
1212
from vllm.distributed import (get_tensor_model_parallel_rank,
1313
get_tensor_model_parallel_world_size)
1414
from vllm.logger import init_logger
15-
from vllm.utils import is_torch_equal_or_newer
1615

1716
__all__ = [
1817
"BasevLLMParameter", "PackedvLLMParameter", "PerTensorScaleParameter",
@@ -115,15 +114,6 @@ def _shard_id_as_int(self, shard_id: Union[str, int]) -> int:
115114

116115
@classmethod
117116
def __torch_function__(cls, func, types, args=(), kwargs=None):
118-
if not is_torch_equal_or_newer("2.8.0"):
119-
logger.warning_once(
120-
"Torch %s detected (<2.8.0): returning NotImplemented in "
121-
"BasevLLMParameter.__torch_function__ to avoid potential "
122-
"TorchDynamo issues.",
123-
torch.__version__,
124-
)
125-
return NotImplemented
126-
127117
if kwargs is None:
128118
kwargs = {}
129119
return super().__torch_function__(func, types, args, kwargs)

0 commit comments

Comments
 (0)