dist/include/ contains the MLX and MLX-C headers needed for CGo compilation. Without these, go-mlx cannot be used as a module dependency (headers not found in module cache). Libraries (dylib/metallib) are still gitignored — users build those locally via cmake. Co-Authored-By: Virgil <virgil@lethean.io>
25 lines
608 B
C
25 lines
608 B
C
// Copyright © 2023 Apple Inc.
|
|
|
|
#pragma once
|
|
|
|
#include "mlx/array.h"
|
|
#include "mlx/backend/cuda/cuda.h"
|
|
#include "mlx/backend/gpu/available.h"
|
|
#include "mlx/backend/metal/metal.h"
|
|
#include "mlx/compile.h"
|
|
#include "mlx/device.h"
|
|
#include "mlx/distributed/distributed.h"
|
|
#include "mlx/distributed/ops.h"
|
|
#include "mlx/einsum.h"
|
|
#include "mlx/export.h"
|
|
#include "mlx/fast.h"
|
|
#include "mlx/fft.h"
|
|
#include "mlx/io.h"
|
|
#include "mlx/linalg.h"
|
|
#include "mlx/memory.h"
|
|
#include "mlx/ops.h"
|
|
#include "mlx/random.h"
|
|
#include "mlx/stream.h"
|
|
#include "mlx/transforms.h"
|
|
#include "mlx/utils.h"
|
|
#include "mlx/version.h"
|