Merge pull request #653 from pengcheng888/issue/652

issue/652- 修改rope函数的测试脚本

test/infinicore/ops/rope.py

@@ -4,6 +4,7 @@ import sys
 sys.path.insert(0, os.path.join(os.path.dirname(__file__), ".."))
 
 
+import torch
 from framework.base import BaseOperatorTest, TensorSpec, TestCase
 from framework.runner import GenericTestRunner
 from framework.utils import is_broadcast
@@ -153,11 +154,13 @@ class OpTest(BaseOperatorTest):
 
         bs, seq_len, num, head_dim = x.shape
 
+        infini_device = x.device
+        torch_device = torch.device(type=infini_device.type, index=infini_device.index)
+
         ## 创建 pos_ids的变量
         cache_position_list = [list(range(0, seq_len)) for i in range(bs)]
-        pos_ids_infini = infinicore.from_list(
-            cache_position_list, dtype=infinicore.int64, device=x.device
-        )
+        pos_ids_torch = torch.tensor(cache_position_list, device=torch_device)
+        pos_ids_infini = infinicore.from_torch(pos_ids_torch)
 
         # 计算
         pos_ids = pos_ids_infini