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

Gemm update (#1518)

Browse Source
This commit is contained in:
Jagrit Digani
2024-10-30 19:30:28 -07:00
committed by GitHub
parent 884af42da2
commit 960e3f0f05
9 changed files with 701 additions and 196 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -181,6 +181,7 @@ Device::Device() {
auto pool = new_scoped_memory_pool();
device_ = load_device();
library_map_ = {{"mlx", load_library(device_)}};
arch_ = std::string(device_->architecture()->name()->utf8String());
}
Device::~Device() {