mirror of
https://github.com/ml-explore/mlx.git
synced 2025-06-26 02:33:21 +08:00
ccf1645995
* extensions start * rope custom op * fix build * docs + rope benchmark * fix test * Add a Metal kernel for RoPE * Fix position of traditional * transform tests * Move rope computation to float and fix tests * Fix the test and a typo * change to fast * fix no metal build --------- Co-authored-by: Angelos Katharopoulos <a_katharopoulos@apple.com>
18 lines
363 B
C
18 lines
363 B
C
// Copyright © 2023 Apple Inc.
|
|
|
|
#pragma once
|
|
|
|
#include "mlx/array.h"
|
|
#include "mlx/backend/metal/metal.h"
|
|
#include "mlx/compile.h"
|
|
#include "mlx/device.h"
|
|
#include "mlx/fast.h"
|
|
#include "mlx/fft.h"
|
|
#include "mlx/io.h"
|
|
#include "mlx/linalg.h"
|
|
#include "mlx/ops.h"
|
|
#include "mlx/random.h"
|
|
#include "mlx/stream.h"
|
|
#include "mlx/transforms.h"
|
|
#include "mlx/utils.h"
|