Skip to content

Commit 1ffefd3

Browse files
add uts for multistream
1 parent 4149c58 commit 1ffefd3

File tree

2 files changed

+4
-4
lines changed

2 files changed

+4
-4
lines changed

tests/ut/multistream/test_metadata.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,12 +1,12 @@
11
from unittest.mock import MagicMock, patch
22

33
import torch
4-
from tests.ut.base import TestBase
4+
from tests.ut.base import TestBase
55
from vllm_ascend.multistream.base import MSEventKey
66
from vllm_ascend.multistream.metadata import (MultiStreamConfig,
77
MultiStreamMetadata,
8-
MultiStreamStepMetadata,
9-
split_micro_batches_tensors)
8+
MultiStreamStepMetadata,
9+
split_micro_batches_tensors)
1010

1111

1212
class TestMetaData(TestBase):

tests/ut/multistream/test_ms_split.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
import unittest
22
from unittest.mock import MagicMock
33

4-
import torch
4+
import torch
55
from vllm_ascend.attention.attention_v1 import AscendAttentionState
66
from vllm_ascend.multistream.base import MSAttentionMetadataSplitConfig
77
from vllm_ascend.multistream.ms_split import (compute_split_seq_index,

0 commit comments

Comments
 (0)