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

Added mlx::core::version() returning std::string(MLX_VERSION) (#1819)

Browse Source 
* Added version.h providing mlx::core::version() returning std::string(MLX_VERSION)

Also, added MLX_VERSION_MAJOR, MLX_VERSION_MINOR, MLX_VERSION_PATCH, MLX_VERSION_NUMERIC, and accompanying functions.

* Added version.h to mlx.h

* Changed version int functions to be constexpr

* Formatting

* Added handling of MLX_VERSION where only the prefix has major.minor.patch format

* Changed version function to be constexpr
This commit is contained in:
Jesper Stemann Andersen
2025-02-20 05:30:19 +01:00
committed by GitHub
parent c707b2b0a6
commit c86422bdd4
4 changed files with 58 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
mlx/export.cpp
View File

@@ -4,9 +4,7 @@
#include "mlx/fast_primitives.h"
#include "mlx/primitives.h"
#include "mlx/utils.h"
#define STRINGIFY(x) #x
#define TOSTRING(x) STRINGIFY(x)
#include "mlx/version.h"
// clang-format off
#define SERIALIZE_PRIMITIVE(primitive, ...) \
@@ -379,7 +377,7 @@ struct PrimitiveFactory {
};
void write_header(Writer& os, int count, bool shapeless) {
serialize(os, std::string(TOSTRING(MLX_VERSION)));
serialize(os, std::string(version()));
serialize(os, count);
serialize(os, shapeless);
}

1
mlx/mlx.h
View File

@@ -19,3 +19,4 @@
#include "mlx/stream.h"
#include "mlx/transforms.h"
#include "mlx/utils.h"
#include "mlx/version.h"

32
mlx/version.h Normal file
View File

@@ -0,0 +1,32 @@
// Copyright © 2023-2024 Apple Inc.
#pragma once
#include <string>
#define STRINGIFY(x) #x
#define TOSTRING(x) STRINGIFY(x)
namespace mlx::core {
constexpr const char* version() {
return TOSTRING(MLX_VERSION);
}
constexpr int version_major() {
return MLX_VERSION_MAJOR;
}
constexpr int version_minor() {
return MLX_VERSION_MINOR;
}
constexpr int version_patch() {
return MLX_VERSION_PATCH;
}
constexpr int version_numeric() {
return MLX_VERSION_NUMERIC;
}
} // namespace mlx::core