[GR-57192] Basic suport for MLX library

PullRequest: graalpython/3435

Author: msimacek

graalpython/com.oracle.graal.python.cext/src/capi.h

@@ -342,6 +342,7 @@ PY_TRUFFLE_TYPE(UnionType_Type, 		"_ctypes.UnionType",		&PyType_Type, sizeof(PyO
 PY_TRUFFLE_TYPE(PyCPointerType_Type,	"PyCPointerType", 		    &PyType_Type, sizeof(PyObject)) \
 PY_TRUFFLE_TYPE(PyCArrayType_Type,		"PyCArrayType", 			&PyType_Type, sizeof(PyObject)) \
 PY_TRUFFLE_TYPE(PyCoro_Type, 			"coroutine", 				&PyType_Type, sizeof(PyCoroObject)) \
+PY_TRUFFLE_TYPE(Py_GenericAliasType,    "types.GenericAlias", 		&PyType_Type, sizeof(PyObject)) \
 /* PyPickleBufferObject (PyObject_HEAD + Py_buffer + PyObject*) is defined within Objects/picklebufobject.c, so its not exposed. */ \
 PY_TRUFFLE_TYPE(PyPickleBuffer_Type, 	"_pickle.PickleBuffer",     &PyType_Type, sizeof(PyPickleBufferObject)) \
 PY_TRUFFLE_TYPE_UNIMPLEMENTED(_PyAIterWrapper_Type) \

graalpython/com.oracle.graal.python.test/src/tests/cpyext/__init__.py

@@ -767,7 +767,7 @@ def CPyExtHeapType(name, bases=(object), code='', slots=None, **kwargs):
         {slots}
     }};
 
-    PyType_Spec spec = {{ "{name}Type", sizeof({name}Object), 0, Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE, slots }};
+    PyType_Spec spec = {{ "{name}", sizeof({name}Object), 0, Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE, slots }};
 
     static PyObject* create(PyObject* unused, PyObject* bases) {{
         {ready_code}

graalpython/com.oracle.graal.python.test/src/tests/cpyext/test_classobject.py

@@ -39,7 +39,7 @@
 
 import sys
 
-from . import CPyExtTestCase, CPyExtFunction
+from . import CPyExtTestCase, CPyExtFunction, CPyExtType, assert_raises, CPyExtHeapType
 
 __dir__ = __file__.rpartition("/")[0]
 
@@ -53,7 +53,6 @@ def foo(self):
 
 
 class TestClassobject(CPyExtTestCase):
-
     testmod = type(sys)("foo")
 
     test_PyMethod_Function = CPyExtFunction(
@@ -79,3 +78,18 @@ class TestClassobject(CPyExtTestCase):
         argspec="O",
         arguments=["PyObject* func"],
     )
+
+    def test_name_qualname(self):
+        TypeWithName = CPyExtType('TypeWithName')
+        assert TypeWithName.__name__ == 'TypeWithName'
+        assert TypeWithName.__qualname__ == 'TypeWithName'
+        assert_raises(TypeError, setattr, '__name__', TypeWithName, "foo")
+        assert_raises(TypeError, setattr, '__qualname__', TypeWithName, "foo")
+
+        HeapTypeWithName = CPyExtHeapType('HeapTypeWithName')
+        assert HeapTypeWithName.__name__ == 'HeapTypeWithName'
+        assert HeapTypeWithName.__qualname__ == 'HeapTypeWithName'
+        HeapTypeWithName.__name__ = 'HeapTypeWithNameRenamed'
+        assert HeapTypeWithName.__name__ == 'HeapTypeWithNameRenamed'
+        HeapTypeWithName.__qualname__ = 'foo.HeapTypeWithNameRenamed'
+        assert HeapTypeWithName.__qualname__ == 'foo.HeapTypeWithNameRenamed'

graalpython/com.oracle.graal.python/src/com/oracle/graal/python/builtins/modules/cext/PythonCextBuiltins.java

@@ -1705,7 +1705,11 @@ private static PythonManagedClass lookupBuiltinTypeWithName(PythonContext contex
                 }
                 Object attribute = PyObjectGetAttr.getUncached().execute(null, moduleObject, toTruffleStringUncached(name));
                 if (attribute != PNone.NO_VALUE) {
-                    clazz = (PythonManagedClass) attribute;
+                    if (attribute instanceof PythonBuiltinClassType builtinType) {
+                        clazz = core.lookupType(builtinType);
+                    } else {
+                        clazz = (PythonManagedClass) attribute;
+                    }
                 }
 
             }

graalpython/com.oracle.graal.python/src/com/oracle/graal/python/builtins/objects/type/TypeBuiltins.java

@@ -0,0 +1,11 @@
+[[add-sources]]
+version = '0.16.3'
+url = 'https://github.com/ml-explore/mlx/archive/refs/tags/v0.16.3.tar.gz'
+
+[[add-sources]]
+version = '0.16.2'
+url = 'https://github.com/ml-explore/mlx/archive/refs/tags/v0.16.2.tar.gz'
+
+[[add-sources]]
+version = '0.16.1'
+url = 'https://github.com/ml-explore/mlx/archive/refs/tags/v0.16.1.tar.gz'

graalpython/lib-graalpython/patches/cmake/cmake.patch

@@ -0,0 +1,3 @@
+[[rules]]
+patch = 'nanobind.patch'
+license = 'BSD-3-Clause'