mirror of
https://github.com/ml-explore/mlx.git
synced 2025-06-25 09:51:17 +08:00
23 lines
432 B
Python
23 lines
432 B
Python
|
# Copyright © 2023-2024 Apple Inc.
|
||
|
|
|
||
|
|
import unittest
|
||
|
|
|
||
|
|
import mlx.core as mx
|
||
|
|
import mlx_tests
|
||
|
|
|
||
|
|
|
||
|
|
class TestCompile(mlx_tests.MLXTestCase):
|
||
|
|
def test_simple_compile(self):
|
||
|
|
def fun(x, y):
|
||
|
|
return x + y
|
||
|
|
|
||
|
|
compiled_fn = mx.compile(fun)
|
||
|
|
compiled_fn = mx.compile(fun)
|
||
|
|
x = mx.array(1.0)
|
||
|
|
y = mx.array(1.0)
|
||
|
|
# out = compiled_fn(x, y)
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
unittest.main()
|