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

moved to subdirectory

Browse Source
This commit is contained in:
dc-dc-dc 2023-12-22 16:05:07 -05:00
parent 13fd7bd721
commit fa093967ec
11 changed files with 14 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
mlx/CMakeLists.txt
View File

@ -8,10 +8,8 @@ target_sources(
${CMAKE_CURRENT_SOURCE_DIR}/fft.cpp
${CMAKE_CURRENT_SOURCE_DIR}/ops.cpp
${CMAKE_CURRENT_SOURCE_DIR}/graph_utils.cpp
${CMAKE_CURRENT_SOURCE_DIR}/load.cpp
${CMAKE_CURRENT_SOURCE_DIR}/primitives.cpp
${CMAKE_CURRENT_SOURCE_DIR}/random.cpp
${CMAKE_CURRENT_SOURCE_DIR}/safetensor.cpp
${CMAKE_CURRENT_SOURCE_DIR}/scheduler.cpp
${CMAKE_CURRENT_SOURCE_DIR}/transforms.cpp
${CMAKE_CURRENT_SOURCE_DIR}/utils.cpp
@ -19,7 +17,7 @@ target_sources(
)
add_subdirectory(${CMAKE_CURRENT_SOURCE_DIR}/backend/common)
add_subdirectory(${CMAKE_CURRENT_SOURCE_DIR}/io)
if (MLX_BUILD_ACCELERATE)
add_subdirectory(${CMAKE_CURRENT_SOURCE_DIR}/backend/accelerate)
else()

2
mlx/backend/common/load.cpp
View File

@ -5,7 +5,7 @@
#include <utility>
#include "mlx/allocator.h"
#include "mlx/load.h"
#include "mlx/io/load.h"
#include "mlx/primitives.h"
namespace mlx::core {

6
mlx/io/CMakeLists.txt Normal file
View File

@ -0,0 +1,6 @@
target_sources(
mlx
PRIVATE
${CMAKE_CURRENT_SOURCE_DIR}/load.cpp
${CMAKE_CURRENT_SOURCE_DIR}/safetensor.cpp
)

2
mlx/load.cpp → mlx/io/load.cpp
View File

@ -6,7 +6,7 @@
#include <limits>
#include <sstream>
#include "mlx/load.h"
#include "mlx/io/load.h"
#include "mlx/ops.h"
#include "mlx/primitives.h"
#include "mlx/utils.h"

0
mlx/load.h → mlx/io/load.h
View File

2
mlx/safetensor.cpp → mlx/io/safetensor.cpp
View File

@ -1,4 +1,4 @@
#include "mlx/safetensor.h"
#include "mlx/io/safetensor.h"
#include <stack>

2
mlx/safetensor.h → mlx/io/safetensor.h
View File

@ -4,7 +4,7 @@
#include <json.hpp>
#include "mlx/load.h"
#include "mlx/io/load.h"
#include "mlx/ops.h"
#include "mlx/primitives.h"

1
mlx/mlx.h
View File

@ -8,7 +8,6 @@
#include "mlx/fft.h"
#include "mlx/ops.h"
#include "mlx/random.h"
#include "mlx/safetensor.h"
#include "mlx/stream.h"
#include "mlx/transforms.h"
#include "mlx/utils.h"

2
mlx/ops.h
View File

@ -7,7 +7,7 @@
#include "array.h"
#include "device.h"
#include "load.h"
#include "io/load.h"
#include "stream.h"
namespace mlx::core {

2
mlx/primitives.h
View File

@ -4,7 +4,7 @@
#include "array.h"
#include "device.h"
#include "load.h"
#include "io/load.h"
#include "stream.h"
#define DEFINE_GRADS() \

2
python/src/load.cpp
View File

@ -10,7 +10,7 @@
#include <unordered_map>
#include <vector>
#include "mlx/load.h"
#include "mlx/io/load.h"
#include "mlx/ops.h"
#include "mlx/utils.h"
#include "python/src/load.h"