There was an error while loading. Please reload this page.
1 parent e07b38b commit 5349f83Copy full SHA for 5349f83
1 file changed
tests/unit_tests/dispatch/test_ascend_rotary.py
@@ -2,13 +2,19 @@
2
3
"""Tests for the Ascend rotary embedding implementation."""
4
5
+import pytest
6
import torch
7
8
from vllm_fl.dispatch.backends.reference.impl.rotary import rotary_embedding_torch
-from vllm_fl.dispatch.backends.vendor.ascend.impl.rotary import rotary_embedding_ascend
9
+
10
+pytest.importorskip("torch_npu")
11
12
13
def test_ascend_rotary_falls_back_for_float32():
14
+ from vllm_fl.dispatch.backends.vendor.ascend.impl.rotary import (
15
+ rotary_embedding_ascend,
16
+ )
17
18
num_tokens = 4
19
num_heads = 2
20
head_size = 8
0 commit comments