mirror of
https://github.com/ollama/ollama.git
synced 2026-04-18 09:03:35 -04:00
10e51c5177
* prefer rocm v6 on windows Avoid building with v7 - more changes are needed * MLX: add header vendoring and remove go build tag This switches to using a vendoring approach for the mlx-c headers so that Go can build without requiring a cmake first. This enables building the new MLX based code by default. Every time cmake runs, the headers are refreshed, so we can easily keep them in sync when we bump mlx versions. Basic Windows and Linux support are verified. * ci: harden for flaky choco repo servers CI sometimes fails due to choco not actually installing cache. Since it just speeds up the build, we can proceed without. * review comments
69 lines
1.7 KiB
C
69 lines
1.7 KiB
C
/* Copyright © 2023-2024 Apple Inc. */
|
|
/* */
|
|
/* This file is auto-generated. Do not edit manually. */
|
|
/* */
|
|
|
|
#ifndef MLX_TRANSFORMS_H
|
|
#define MLX_TRANSFORMS_H
|
|
|
|
#include <stdbool.h>
|
|
#include <stdint.h>
|
|
#include <stdio.h>
|
|
|
|
#include "mlx/c/array.h"
|
|
#include "mlx/c/closure.h"
|
|
#include "mlx/c/distributed_group.h"
|
|
#include "mlx/c/io_types.h"
|
|
#include "mlx/c/map.h"
|
|
#include "mlx/c/stream.h"
|
|
#include "mlx/c/string.h"
|
|
#include "mlx/c/vector.h"
|
|
|
|
#ifdef __cplusplus
|
|
extern "C" {
|
|
#endif
|
|
|
|
/**
|
|
* \defgroup transforms Transform operations
|
|
*/
|
|
/**@{*/
|
|
|
|
int mlx_async_eval(const mlx_vector_array outputs);
|
|
int mlx_checkpoint(mlx_closure* res, const mlx_closure fun);
|
|
int mlx_custom_function(
|
|
mlx_closure* res,
|
|
const mlx_closure fun,
|
|
const mlx_closure_custom fun_vjp /* may be null */,
|
|
const mlx_closure_custom_jvp fun_jvp /* may be null */,
|
|
const mlx_closure_custom_vmap fun_vmap /* may be null */);
|
|
int mlx_custom_vjp(
|
|
mlx_closure* res,
|
|
const mlx_closure fun,
|
|
const mlx_closure_custom fun_vjp);
|
|
int mlx_eval(const mlx_vector_array outputs);
|
|
int mlx_jvp(
|
|
mlx_vector_array* res_0,
|
|
mlx_vector_array* res_1,
|
|
const mlx_closure fun,
|
|
const mlx_vector_array primals,
|
|
const mlx_vector_array tangents);
|
|
int mlx_value_and_grad(
|
|
mlx_closure_value_and_grad* res,
|
|
const mlx_closure fun,
|
|
const int* argnums,
|
|
size_t argnums_num);
|
|
int mlx_vjp(
|
|
mlx_vector_array* res_0,
|
|
mlx_vector_array* res_1,
|
|
const mlx_closure fun,
|
|
const mlx_vector_array primals,
|
|
const mlx_vector_array cotangents);
|
|
|
|
/**@}*/
|
|
|
|
#ifdef __cplusplus
|
|
}
|
|
#endif
|
|
|
|
#endif
|