core/cli
8
0
Fork 0
Code Issues 1 Pull requests Projects Releases Packages Wiki Activity Actions 1

fix: correct mlx_closure_new_func_payload signature for mlx-c v0.4.1

Browse source 
Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
This commit is contained in:
Claude 2026-02-16 01:41:07 +00:00 • committed by Snider
parent 8ee0c4bc4e
commit c25e1a633c
1 changed files with 6 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

11
pkg/mlx/compile.go
View file

@ -6,10 +6,10 @@ package mlx
#include "mlx/c/mlx.h" #include "mlx/c/mlx.h"
// Callback for compiled functions. // Callback for compiled functions.
extern void goCompiledFunc(mlx_vector_array inputs, mlx_vector_array outputs, void *payload); extern int goCompiledFunc(mlx_vector_array *outputs, const mlx_vector_array inputs, void *payload);
static mlx_closure new_closure(void *payload) { static mlx_closure new_closure(void *payload) {
return mlx_closure_new_func_payload(&goCompiledFunc, payload); return mlx_closure_new_func_payload(&goCompiledFunc, payload, NULL);
} }
*/ */
import "C" import "C"
@ -29,11 +29,11 @@ type CompiledFunc struct {
var compiledFuncs sync.Map var compiledFuncs sync.Map
//export goCompiledFunc //export goCompiledFunc
func goCompiledFunc(inputs C.mlx_vector_array, outputs C.mlx_vector_array, payload unsafe.Pointer) { func goCompiledFunc(outputs *C.mlx_vector_array, inputs C.mlx_vector_array, payload unsafe.Pointer) C.int {
id := uintptr(payload) id := uintptr(payload)
fnI, ok := compiledFuncs.Load(id) fnI, ok := compiledFuncs.Load(id)
if !ok { if !ok {
return return 1
} }
fn := fnI.(func([]*Array) []*Array) fn := fnI.(func([]*Array) []*Array)
@ -51,8 +51,9 @@ func goCompiledFunc(inputs C.mlx_vector_array, outputs C.mlx_vector_array, paylo
// Set outputs // Set outputs
for _, out := range goOutputs { for _, out := range goOutputs {
C.mlx_vector_array_append_value(outputs, out.ctx) C.mlx_vector_array_append_value(*outputs, out.ctx)
} }
return 0
} }
var nextID uintptr var nextID uintptr