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

Search mlx.metallib in macOS framework

Browse Source
This commit is contained in:
Hanyu Deng 2025-04-10 18:17:12 +08:00 committed by Angelos Katharopoulos
parent e8ac6bd2f5
commit f360823cee
2 changed files with 22 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
mlx/backend/metal/device.cpp
View File

@ -1,5 +1,6 @@
// Copyright © 2023-2024 Apple Inc. // Copyright © 2023-2024 Apple Inc.
#include <array>
#include <cstdlib> #include <cstdlib>
#include <sstream> #include <sstream>
@ -17,6 +18,22 @@
namespace mlx::core::metal { namespace mlx::core::metal {
std::string search_colocated_mtllib(
const fs::path& dli_path,
const std::string& lib_name) {
auto dir_name = dli_path.parent_path();
std::array search_list{
dir_name / lib_name,
dir_name / "Resources" / lib_name // in macOS framework
};
for (const auto& lib_path : search_list) {
if (fs::exists(lib_path)) {
return lib_path.c_str();
}
}
return "";
}
namespace { namespace {
constexpr const char* default_mtllib_path = METAL_PATH; constexpr const char* default_mtllib_path = METAL_PATH;

7
mlx/backend/metal/device.h
View File

@ -19,6 +19,10 @@ namespace fs = std::filesystem;
namespace mlx::core::metal { namespace mlx::core::metal {
std::string search_colocated_mtllib(
const fs::path& dli_path,
const std::string& lib_name);
// Note, this function must be left inline in a header so that it is not // Note, this function must be left inline in a header so that it is not
// dynamically linked. // dynamically linked.
inline std::string get_colocated_mtllib_path(const std::string& lib_name) { inline std::string get_colocated_mtllib_path(const std::string& lib_name) {
@ -28,8 +32,7 @@ inline std::string get_colocated_mtllib_path(const std::string& lib_name) {
int success = dladdr((void*)get_colocated_mtllib_path, &info); int success = dladdr((void*)get_colocated_mtllib_path, &info);
if (success) { if (success) {
auto mtllib = fs::path(info.dli_fname).remove_filename() / lib_ext; mtllib_path = search_colocated_mtllib(fs::path(info.dli_fname), lib_ext);
mtllib_path = mtllib.c_str();
} }
return mtllib_path; return mtllib_path;