mirror of
https://github.com/ml-explore/mlx.git
synced 2025-07-15 04:51:13 +08:00

* expose residency sets as wire/unwire * returns wired size * fix * runtime support check * fix os check * fix test * fix no metal build * docs * nit * nits in docs * nits
64 lines
1.3 KiB
C++
64 lines
1.3 KiB
C++
// Copyright © 2023-2024 Apple Inc.
|
|
|
|
#include <stdexcept>
|
|
|
|
#include "mlx/backend/metal/metal.h"
|
|
#include "mlx/backend/metal/metal_impl.h"
|
|
namespace mlx::core::metal {
|
|
|
|
bool is_available() {
|
|
return false;
|
|
}
|
|
|
|
void new_stream(Stream) {}
|
|
|
|
std::unique_ptr<void, std::function<void(void*)>> new_scoped_memory_pool() {
|
|
return nullptr;
|
|
}
|
|
|
|
std::function<void()> make_task(array, bool) {
|
|
throw std::runtime_error(
|
|
"[metal::make_task] Cannot make GPU task without metal backend");
|
|
}
|
|
|
|
std::function<void()> make_synchronize_task(
|
|
Stream,
|
|
std::shared_ptr<std::promise<void>>) {
|
|
throw std::runtime_error(
|
|
"[metal::make_synchronize_task] Cannot synchronize GPU"
|
|
" without metal backend");
|
|
}
|
|
|
|
// No-ops when Metal is not available.
|
|
size_t get_active_memory() {
|
|
return 0;
|
|
}
|
|
size_t get_peak_memory() {
|
|
return 0;
|
|
}
|
|
void reset_peak_memory() {}
|
|
size_t get_cache_memory() {
|
|
return 0;
|
|
}
|
|
size_t set_memory_limit(size_t, bool) {
|
|
return 0;
|
|
}
|
|
size_t set_cache_limit(size_t) {
|
|
return 0;
|
|
}
|
|
size_t set_wired_limit(size_t) {
|
|
return 0;
|
|
}
|
|
|
|
void start_capture(std::string) {}
|
|
void stop_capture() {}
|
|
void clear_cache() {}
|
|
|
|
std::unordered_map<std::string, std::variant<std::string, size_t>>
|
|
device_info() {
|
|
throw std::runtime_error(
|
|
"[metal::device_info] Cannot get device info without metal backend");
|
|
};
|
|
|
|
} // namespace mlx::core::metal
|