maxframe 1.3.0__cp310-cp310-win32.whl → 2.0.0__cp310-cp310-win32.whl

maxframe/codegen/spe/tensor/tests/test_datasource.py

@@ -0,0 +1,99 @@
+# Copyright 1999-2025 Alibaba Group Holding Ltd.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#      http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+import pytest
+
+from ..... import tensor as mt
+from ...core import SPECodeContext
+from ..datasource import (
+    TensorArangeAdapter,
+    TensorEmptyAdapter,
+    TensorFullAdapter,
+    TensorOnesZerosAdapter,
+    TensorOnesZerosLikeAdapter,
+    TensorTriArrayAdapter,
+)
+
+
+def test_arange():
+    s1 = mt.arange(10)
+    adapter = TensorArangeAdapter()
+    context = SPECodeContext()
+    results = adapter.generate_code(s1.op, context)
+    expected_results = [
+        "var_0 = np.arange(const_0, const_1, const_2, dtype=np.dtype('int64'))"
+    ]
+    assert results == expected_results
+    assert [context.constants[f"const_{i}"] for i in range(3)] == [0, 10, 1]
+
+
+def test_empty():
+    s1 = mt.empty((100, 100))
+    adapter = TensorEmptyAdapter()
+    context = SPECodeContext()
+    results = adapter.generate_code(s1.op, context)
+    expected_results = ["var_0 = np.empty((100, 100), dtype=const_0, order='C')"]
+    assert results == expected_results
+
+
+def test_full():
+    s1 = mt.full((100, 100), 1)
+    adapter = TensorFullAdapter()
+    context = SPECodeContext()
+    results = adapter.generate_code(s1.op, context)
+    expected_results = [
+        "var_0 = np.full((100, 100), 1, dtype=np.dtype('int64'), order='C')"
+    ]
+    assert results == expected_results
+
+
+@pytest.mark.parametrize("method", ["triu", "tril"])
+def test_tri_array(method):
+    s1 = mt.random.rand(20, 20)
+    adapter = TensorTriArrayAdapter()
+    s3 = getattr(mt, method)(s1, k=1)
+    context = SPECodeContext()
+    results = adapter.generate_code(s3.op, context)
+    expected_results = [f"var_1 = np.{method}(var_0, k=1)"]
+    assert results == expected_results
+
+
+@pytest.mark.parametrize("method", ["ones", "zeros"])
+def test_ones_zeros(method):
+    s1 = getattr(mt, method)((1000, 5), order="F")
+    adapter = TensorOnesZerosAdapter()
+    context = SPECodeContext()
+    results = adapter.generate_code(s1.op, context)
+    expected_results = [f"var_0 = np.{method}((1000, 5), dtype=const_0, order='F')"]
+    assert results == expected_results
+
+    s2 = getattr(mt, method)((1000,), order="F")
+    adapter = TensorOnesZerosAdapter()
+    context = SPECodeContext()
+    results = adapter.generate_code(s2.op, context)
+    expected_results = [f"var_0 = np.{method}((1000,), dtype=const_0, order='F')"]
+    assert results == expected_results
+
+
+@pytest.mark.parametrize("method", ["ones_like", "zeros_like"])
+def test_ones_zeros_like(method):
+    data = mt.tensor([[2, 0, 0], [0, 0, 2]])
+    s1 = getattr(mt, method)(data, order="F")
+    adapter = TensorOnesZerosLikeAdapter()
+    context = SPECodeContext()
+    results = adapter.generate_code(s1.op, context)
+    expected_results = [
+        f"var_1 = np.{method}(var_0, dtype=const_0, order='F', shape=(2, 3))"
+    ]
+    assert results == expected_results

maxframe/codegen/spe/tensor/tests/test_extensions.py

@@ -0,0 +1,37 @@
+# Copyright 1999-2025 Alibaba Group Holding Ltd.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#      http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+from ..... import tensor as mt
+from ...core import SPECodeContext
+from ..extensions import TensorApplyChunkAdapter
+
+
+def test_tensor_apply_chunk():
+    s1 = mt.random.rand(10, 20, 30)
+
+    adapter = TensorApplyChunkAdapter()
+
+    result = s1.mf.apply_chunk(lambda x: x + 1)
+    context = SPECodeContext()
+    results = adapter.generate_code(result.op, context)
+    expected_results = [f"var_1 = {context.get_udfs()[0].name}(var_0)"]
+    assert results == expected_results
+
+    result = s1.mf.apply_chunk(lambda x, chunk_index: x + 1, with_chunk_index=True)
+    context = SPECodeContext()
+    results = adapter.generate_code(result.op, context)
+    expected_results = [
+        f"var_1 = {context.get_udfs()[0].name}(var_0, chunk_index=(0, 0, 0))"
+    ]
+    assert results == expected_results

maxframe/codegen/spe/tensor/tests/test_indexing.py

@@ -0,0 +1,44 @@
+# Copyright 1999-2025 Alibaba Group Holding Ltd.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#      http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+from ..... import tensor as mt
+from ...core import SPECodeContext
+from ..indexing import TensorIndexAdapter, TensorIndexSetValueAdapter
+
+
+def test_getitem():
+    s1 = mt.random.rand(10, 20, 30)
+    result = s1[:, 1, 2:3]
+
+    adapter = TensorIndexAdapter()
+    context = SPECodeContext()
+    results = adapter.generate_code(result.op, context)
+    expected_results = [
+        "var_0 = var_1[tuple([slice(None, None, None), 1, slice(2, 3, None)])]"
+    ]
+    assert results == expected_results
+
+
+def test_setitem():
+    s1 = mt.random.rand(10, 20, 30)
+    s1[:, 1, 2:3] = 1.0
+
+    adapter = TensorIndexSetValueAdapter()
+    context = SPECodeContext()
+    results = adapter.generate_code(s1.op, context)
+    expected_results = [
+        "var_0 = var_1.copy()",
+        "var_0[tuple((slice(None, None, None), 1, slice(2, 3, None)))] = 1.0",
+    ]
+    assert results == expected_results

maxframe/codegen/spe/tensor/tests/test_linalg.py

@@ -0,0 +1,38 @@
+# Copyright 1999-2025 Alibaba Group Holding Ltd.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#      http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+from ..... import tensor as mt
+from ...core import SPECodeContext
+from ..linalg import TensorQRAdapter, TensorTensorDotAdapter
+
+
+def test_qr():
+    a = mt.random.randn(9, 6)
+    adapter = TensorQRAdapter()
+    q, r = mt.linalg.qr(a)
+    context = SPECodeContext()
+    results = adapter.generate_code(q.op, context)
+    expected_results = ["var_0, var_1 = spl.qr(var_2)"]
+    assert results == expected_results
+
+
+def test_tensordot():
+    s1 = mt.arange(60).reshape(3, 4, 5)
+    s2 = mt.arange(24).reshape(4, 3, 2)
+    adapter = TensorTensorDotAdapter()
+    s3 = mt.tensordot(s1, s2, axes=([1, 0], [0, 1]))
+    context = SPECodeContext()
+    results = adapter.generate_code(s3.op, context)
+    expected_results = ["var_2 = npl.tensordot(var_0, var_1, ((1, 0), (0, 1)))"]
+    assert results == expected_results

maxframe/codegen/spe/tensor/tests/test_merge.py

@@ -0,0 +1,28 @@
+# Copyright 1999-2025 Alibaba Group Holding Ltd.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#      http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+from ..... import tensor as mt
+from ...core import SPECodeContext
+from ..merge import TensorConcatenateAdapter
+
+
+def test_concatenate():
+    s1 = mt.random.rand(1000, 5)
+    s2 = mt.random.rand(1000, 5)
+    adapter = TensorConcatenateAdapter()
+    s3 = mt.concatenate([s1, s2], axis=1)
+    context = SPECodeContext()
+    results = adapter.generate_code(s3.op, context)
+    expected_results = ["var_2 = np.concatenate((var_0, var_1), axis=1)"]
+    assert results == expected_results

maxframe/codegen/spe/tensor/tests/test_misc.py

@@ -0,0 +1,94 @@
+# Copyright 1999-2025 Alibaba Group Holding Ltd.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#      http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+from ..... import tensor as mt
+from ...core import SPECodeContext
+from ..misc import (
+    TensorIsInAdapter,
+    TensorSqueezeAdapter,
+    TensorTransposeAdapter,
+    TensorUniqueAdapter,
+)
+
+
+def test_isin():
+    result = mt.isin(2 * mt.arange(4).reshape((2, 2)), [1, 2, 4, 8])
+
+    adapter = TensorIsInAdapter()
+    context = SPECodeContext()
+    results = adapter.generate_code(result.op, context)
+    expected_results = [
+        "var_0 = np.isin(var_1, var_2, assume_unique=False, invert=False)"
+    ]
+    assert results == expected_results
+
+
+def test_transpose():
+    result = mt.transpose(mt.ones((1, 2, 3)), (1, 0, 2))
+
+    adapter = TensorTransposeAdapter()
+    context = SPECodeContext()
+    results = adapter.generate_code(result.op, context)
+    expected_results = ["var_1 = np.transpose(var_0, axes=[1, 0, 2])"]
+    assert results == expected_results
+
+
+def test_squeeze():
+    result = mt.squeeze(mt.ones((1, 2, 3)), axis=0)
+
+    adapter = TensorSqueezeAdapter()
+    context = SPECodeContext()
+    results = adapter.generate_code(result.op, context)
+    expected_results = ["var_1 = np.squeeze(var_0, axis=(0,))"]
+    assert results == expected_results
+
+
+def test_unique():
+    tensor = mt.unique([1, 1])
+    adapter = TensorUniqueAdapter()
+    context = SPECodeContext()
+    results = adapter.generate_code(tensor.op, context)
+    expected_results = [
+        "var_1 = np.unique(var_0, return_index=False, return_inverse=False, return_counts=False, axis=0)"
+    ]
+    assert results == expected_results
+
+    tensor = mt.unique([1, 1], return_index=True)
+    context = SPECodeContext()
+    results = adapter.generate_code(tensor[0].op, context)
+    expected_results = [
+        "var_1, var_2 = "
+        "np.unique(var_0, return_index=True, return_inverse=False, return_counts=False, axis=0)"
+    ]
+    assert results == expected_results
+
+    tensor = mt.unique([1, 1], return_index=True, return_inverse=True)
+    context = SPECodeContext()
+    results = adapter.generate_code(tensor[0].op, context)
+    expected_results = [
+        "var_1, var_2, var_3 = "
+        "np.unique(var_0, return_index=True, return_inverse=True, return_counts=False, axis=0)"
+    ]
+    assert results == expected_results
+
+    tensor = mt.unique(
+        [1, 1], return_index=True, return_inverse=True, return_counts=True
+    )
+    context = SPECodeContext()
+    results = adapter.generate_code(tensor[0].op, context)
+    expected_results = [
+        "var_1, var_2, var_3, var_4 = "
+        "np.unique(var_0, return_index=True, return_inverse=True, return_counts=True, axis=0)"
+    ]
+    assert results == expected_results

maxframe/codegen/spe/tensor/tests/test_random.py

@@ -0,0 +1,55 @@
+# Copyright 1999-2025 Alibaba Group Holding Ltd.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#      http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+from ..... import tensor as mt
+from ...core import SPECodeContext
+from ..random import TensorRandAdapter
+
+
+def test_rand_multi_dimension():
+    s1 = mt.random.rand(1000, 5)
+    adapter = TensorRandAdapter()
+    context = SPECodeContext()
+    results = adapter.generate_code(s1.op, context)
+    expected_results = ["var_0 = np.random.rand(1000, 5)"]
+    assert results == expected_results
+
+
+def test_rand_no_dimension():
+    s1 = mt.random.rand()
+    adapter = TensorRandAdapter()
+    context = SPECodeContext()
+    results = adapter.generate_code(s1.op, context)
+    expected_results = ["var_0 = np.random.rand()"]
+    assert results == expected_results
+
+
+def test_rand_backward_compatibility_multi_dimension():
+    s1 = mt.random.rand(10, 5, 20)
+    s1.op.size = None
+    adapter = TensorRandAdapter()
+    context = SPECodeContext()
+    results = adapter.generate_code(s1.op, context)
+    expected_results = ["var_0 = np.random.rand(10, 5, 20)"]
+    assert results == expected_results
+
+
+def test_rand_backward_compatibility_no_dimension():
+    s1 = mt.random.rand()
+    s1.op.size = None
+    adapter = TensorRandAdapter()
+    context = SPECodeContext()
+    results = adapter.generate_code(s1.op, context)
+    expected_results = ["var_0 = np.random.rand()"]
+    assert results == expected_results

maxframe/codegen/spe/tensor/tests/test_reduction.py

@@ -0,0 +1,65 @@
+# Copyright 1999-2025 Alibaba Group Holding Ltd.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#      http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+from ..... import tensor as mt
+from ...core import SPECodeContext
+from ..reduction import (
+    TensorCumReductionAdapter,
+    TensorReductionAdapter,
+    TensorVarAdapter,
+)
+
+
+def test_reduction():
+    s1 = mt.random.rand(1000, 5)
+    result = mt.argmax(s1, axis=1)
+
+    adapter = TensorReductionAdapter()
+    context = SPECodeContext()
+    results = adapter.generate_code(result.op, context)
+    expected_results = ["var_1 = np.argmax(var_0, axis=1)"]
+    assert results == expected_results
+
+
+def test_cum_reduction():
+    s1 = mt.random.rand(1000, 5)
+    result = mt.cumsum(s1, axis=1)
+
+    adapter = TensorCumReductionAdapter()
+    context = SPECodeContext()
+    results = adapter.generate_code(result.op, context)
+    expected_results = ["var_1 = np.cumsum(var_0, axis=1)"]
+    assert results == expected_results
+
+
+def test_var():
+    s1 = mt.random.rand(1000, 5)
+    result = mt.var(s1, ddof=0.1)
+
+    adapter = TensorVarAdapter()
+    context = SPECodeContext()
+    results = adapter.generate_code(result.op, context)
+    expected_results = ["var_1 = np.var(var_0, ddof=0.1)"]
+    assert results == expected_results
+
+
+def test_nanvar():
+    s1 = mt.random.rand(1000, 5)
+    result = mt.nanvar(s1, ddof=0.1)
+
+    adapter = TensorVarAdapter()
+    context = SPECodeContext()
+    results = adapter.generate_code(result.op, context)
+    expected_results = ["var_1 = np.nanvar(var_0, ddof=0.1)"]
+    assert results == expected_results

maxframe/codegen/spe/tensor/tests/test_reshape.py

@@ -0,0 +1,39 @@
+# Copyright 1999-2025 Alibaba Group Holding Ltd.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#      http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+from ..... import tensor as mt
+from ...core import SPECodeContext
+from ..reshape import TensorReshapeAdapter
+
+
+def test_reshape_C():
+    t1 = mt.arange(6).reshape((3, 2))
+    t2 = mt.reshape(t1, (2, 3))
+    adapter = TensorReshapeAdapter()
+    context = SPECodeContext()
+    context.get_input_tileable_variable(t1.op.outputs[0])
+    results = adapter.generate_code(t2.op, context)
+    expected_results = ["var_1 = np.reshape(var_0, (2, 3), 'C')"]
+    assert results == expected_results
+
+
+def test_reshape_F():
+    t1 = mt.arange(6).reshape((3, 2))
+    t2 = mt.reshape(t1, (2, 3), "F")
+    adapter = TensorReshapeAdapter()
+    context = SPECodeContext()
+    context.get_input_tileable_variable(t1.op.outputs[0])
+    results = adapter.generate_code(t2.op, context)
+    expected_results = ["var_1 = np.reshape(var_0, (2, 3), 'F')"]
+    assert results == expected_results

maxframe/codegen/spe/tensor/tests/test_sort.py

@@ -0,0 +1,49 @@
+# Copyright 1999-2025 Alibaba Group Holding Ltd.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#      http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+from ..... import tensor as mt
+from ...core import SPECodeContext
+from ..sort import TensorSortAdapter
+
+
+def test_sort():
+    t1 = mt.array([1, 4, 5, 3])
+    t2 = mt.sort(t1, stable=True)
+    adapter = TensorSortAdapter()
+    context = SPECodeContext()
+    context.get_input_tileable_variable(t1.op.outputs[0])
+    results = adapter.generate_code(t2.op, context)
+    expected_results = ["var_1 = np.sort(var_0, axis=0, kind='STABLE', order=None)"]
+    assert results == expected_results
+
+    t2, t3 = mt.sort(t1, stable=True, return_index=True)
+    adapter = TensorSortAdapter()
+    context = SPECodeContext()
+    context.get_input_tileable_variable(t1.op.outputs[0])
+    results = adapter.generate_code(t2.op, context)
+    expected_results = [
+        "var_1 = np.sort(var_0, axis=0, kind='STABLE', order=None)",
+        "var_2 = np.argsort(var_0, axis=0, kind='STABLE', order=None)",
+    ]
+    assert results == expected_results
+
+    t2 = mt.argsort(t1, stable=True)
+    adapter = TensorSortAdapter()
+    context = SPECodeContext()
+    context.get_input_tileable_variable(t1.op.outputs[0])
+    results = adapter.generate_code(t2.op, context)
+    expected_results = [
+        "var_1 = np.argsort(var_0, axis=0, kind='STABLE', order=None)",
+    ]
+    assert results == expected_results

maxframe/codegen/spe/tensor/tests/test_special.py

@@ -0,0 +1,28 @@
+# Copyright 1999-2025 Alibaba Group Holding Ltd.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#      http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+from ..... import tensor as mt
+from .....tensor.special import expit
+from ...core import SPECodeContext
+from ..special import TensorSpecialUnaryOpAdapter
+
+
+def test_special_unary():
+    result = expit(mt.random.rand(10, 10))
+
+    adapter = TensorSpecialUnaryOpAdapter()
+    context = SPECodeContext()
+    results = adapter.generate_code(result.op, context)
+    expected_results = ["var_1 = spspecial.expit(var_0)"]
+    assert results == expected_results

maxframe/codegen/spe/tensor/tests/test_statistics.py

@@ -0,0 +1,29 @@
+# Copyright 1999-2025 Alibaba Group Holding Ltd.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#      http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+from ..... import tensor as mt
+from ...core import SPECodeContext
+from ..statistics import TensorBinCountAdapter
+
+
+def test_bincount():
+    w = mt.array([0.3, 0.5, 0.2, 0.7, 1.0, -0.6])  # weights
+    x = mt.array([0, 1, 1, 2, 2, 2])
+    result = mt.bincount(x, weights=w)
+
+    adapter = TensorBinCountAdapter()
+    context = SPECodeContext()
+    results = adapter.generate_code(result.op, context)
+    expected_results = ["var_0 = np.bincount(var_1, weights=var_2, minlength=0)"]
+    assert results == expected_results

maxframe/codegen/spe/tests/__init__.py

@@ -0,0 +1,13 @@
+# Copyright 1999-2025 Alibaba Group Holding Ltd.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#      http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.

maxframe/codegen/spe/tests/test_remote.py

@@ -0,0 +1,29 @@
+# Copyright 1999-2025 Alibaba Group Holding Ltd.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#      http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+from .... import remote as mr
+from ..core import SPECodeContext
+from ..remote import RemoteFunctionAdapter
+
+
+def test_remote():
+    def f(v1, v2):
+        return v1 + v2
+
+    v0 = mr.spawn(f, (2, 3))
+    v1 = mr.spawn(f, (v0, 10))
+    context = SPECodeContext()
+    results = RemoteFunctionAdapter().generate_code(v1.op, context)
+    assert callable(context.constants["const_0"])
+    assert results[0] == "var_1 = const_0(var_0, 10)"