File tree Expand file tree Collapse file tree 2 files changed +4
-4
lines changed Expand file tree Collapse file tree 2 files changed +4
-4
lines changed Original file line number Diff line number Diff line change 1
1
from unittest .mock import MagicMock , patch
2
2
3
3
import torch
4
- from tests .ut .base import TestBase
4
+ from tests .ut .base import TestBase
5
5
from vllm_ascend .multistream .base import MSEventKey
6
6
from vllm_ascend .multistream .metadata import (MultiStreamConfig ,
7
7
MultiStreamMetadata ,
8
- MultiStreamStepMetadata ,
9
- split_micro_batches_tensors )
8
+ MultiStreamStepMetadata ,
9
+ split_micro_batches_tensors )
10
10
11
11
12
12
class TestMetaData (TestBase ):
Original file line number Diff line number Diff line change 1
1
import unittest
2
2
from unittest .mock import MagicMock
3
3
4
- import torch
4
+ import torch
5
5
from vllm_ascend .attention .attention_v1 import AscendAttentionState
6
6
from vllm_ascend .multistream .base import MSAttentionMetadataSplitConfig
7
7
from vllm_ascend .multistream .ms_split import (compute_split_seq_index ,
You can’t perform that action at this time.
0 commit comments