adapted to ms2.7

Author: QiuYuFong

examples/opensora_hpcai/opensora/models/layers/operation_selector.py

@@ -1,6 +1,6 @@
 import mindspore as ms
 from mindspore import mint, ops
-from mindspore.ops.function.array_func import chunk_ext, repeat_interleave_ext
+from mindspore.ops.function.array_func import repeat_interleave_ext
 
 use_dynamic_ops = False
 
@@ -66,7 +66,7 @@ def get_chunk_op():
     if (mode == 0) and (not check_dynamic_mode()):
         return ops.chunk
     else:
-        return chunk_ext
+        return mint.chunk
 
 
 def get_split_op():

examples/opensora_hpcai/opensora/models/layers/rotary_embedding.py

@@ -12,15 +12,14 @@
 import numpy as np
 
 import mindspore as ms
-from mindspore import Parameter, Tensor, dtype, nn, ops
-from mindspore.ops.function.array_func import chunk_ext as chunk
+from mindspore import Parameter, Tensor, dtype, nn, ops, mint
 
 from .operation_selector import get_repeat_interleave_op
 
 
 def rotate_half(x: Tensor) -> Tensor:
     x = x.reshape(x.shape[:-1] + (-1, 2))  # ... (d r) -> ... d r, r = 2
-    x1, x2 = chunk(x, 2, -1)
+    x1, x2 = mint.chunk(x, 2, -1)
     x = ops.concat((-x2, x1), axis=-1)
     return x.reshape(x.shape[:-2] + (-1,))  # '... d r -> ... (d r)'