zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-10-25 12:48:14 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Separate fast ops and primitives (#699)

Browse Source
This commit is contained in:
Awni Hannun
2024-02-16 19:16:39 -08:00
committed by GitHub
parent bf7cd29970
commit c3965fc5ee
6 changed files with 74 additions and 70 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
mlx/backend/common/rope.cpp
View File

@@ -1,7 +1,6 @@
// Copyright © 2023-2024 Apple Inc.
#include "mlx/fast.h"
#include "mlx/primitives.h"
#include "mlx/fast_primitives.h"
namespace mlx::core::fast {

3
mlx/backend/metal/rope.cpp
View File

@@ -1,8 +1,7 @@
// Copyright © 2023-2024 Apple Inc.
#include "mlx/backend/metal/utils.h"
#include "mlx/fast.h"
#include "mlx/primitives.h"
#include "mlx/fast_primitives.h"
namespace mlx::core::fast {

2
mlx/backend/no_metal/primitives.cpp
View File

@@ -1,7 +1,7 @@
// Copyright © 2023-2024 Apple Inc.
#include "mlx/primitives.h"
#include "mlx/fast.h"
#include "mlx/fast_primitives.h"
#define NO_GPU_MULTI(func) \
void func::eval_gpu( \