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 bc28aad526
commit c6597691bb
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"
// 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) {
return mlx_closure_new_func_payload(&goCompiledFunc, payload);
return mlx_closure_new_func_payload(&goCompiledFunc, payload, NULL);
}
*/
import "C"
@ -29,11 +29,11 @@ type CompiledFunc struct {
var compiledFuncs sync.Map
//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)
fnI, ok := compiledFuncs.Load(id)
if !ok {
return
return 1
}
fn := fnI.(func([]*Array) []*Array)
@ -51,8 +51,9 @@ func goCompiledFunc(inputs C.mlx_vector_array, outputs C.mlx_vector_array, paylo
// Set outputs
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