Unverified Commit 68feb76a authored by Roy Wang's avatar Roy Wang Committed by GitHub
Browse files

[Misc] Replace deprecated interface seed_everything (#33474)


Signed-off-by: default avataresmeetu <jasonailu87@gmail.com>
parent 4cb59dea
...@@ -14,7 +14,7 @@ from vllm._custom_ops import ( ...@@ -14,7 +14,7 @@ from vllm._custom_ops import (
) )
from vllm.platforms import CpuArchEnum, current_platform from vllm.platforms import CpuArchEnum, current_platform
from vllm.utils.argparse_utils import FlexibleArgumentParser from vllm.utils.argparse_utils import FlexibleArgumentParser
from vllm.utils.torch_utils import STR_DTYPE_TO_TORCH_DTYPE from vllm.utils.torch_utils import STR_DTYPE_TO_TORCH_DTYPE, set_random_seed
from vllm.v1.attention.backends.cpu_attn import CPUAttentionBackend, _get_attn_isa from vllm.v1.attention.backends.cpu_attn import CPUAttentionBackend, _get_attn_isa
...@@ -58,7 +58,7 @@ def main( ...@@ -58,7 +58,7 @@ def main(
seed: int = 0, seed: int = 0,
iters: int = 20, iters: int = 20,
) -> None: ) -> None:
current_platform.seed_everything(seed) set_random_seed(seed)
num_seqs = len(seq_lens) num_seqs = len(seq_lens)
query_lens = [x[0] for x in seq_lens] query_lens = [x[0] for x in seq_lens]
kv_lens = [x[1] for x in seq_lens] kv_lens = [x[1] for x in seq_lens]
......
...@@ -7,8 +7,8 @@ import time ...@@ -7,8 +7,8 @@ import time
import numpy as np import numpy as np
import torch import torch
from vllm.platforms import current_platform
from vllm.utils.argparse_utils import FlexibleArgumentParser from vllm.utils.argparse_utils import FlexibleArgumentParser
from vllm.utils.torch_utils import set_random_seed
# Check if CPU MoE operations are available # Check if CPU MoE operations are available
try: try:
...@@ -41,7 +41,7 @@ def main( ...@@ -41,7 +41,7 @@ def main(
seed: int = 0, seed: int = 0,
iters: int = 20, iters: int = 20,
) -> None: ) -> None:
current_platform.seed_everything(seed) set_random_seed(seed)
# up_dim = 2 * intermediate_size for gate + up projection # up_dim = 2 * intermediate_size for gate + up projection
up_dim = 2 * intermediate_size up_dim = 2 * intermediate_size
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment