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

Allow dynamic ops per buffer based on dispatches and memory (#1864)

Browse Source 
* Allow dynamic ops per buffer based on dispatches and memory

* add initial arch values
This commit is contained in:
Awni Hannun
2025-02-13 19:18:22 -08:00
committed by GitHub
parent 9733e16496
commit 7aea5b1895
5 changed files with 62 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
mlx/backend/metal/matmul.cpp
View File

@@ -109,7 +109,7 @@ std::tuple<bool, int64_t, array> check_transpose(
///////////////////////////////////////////////////////////////////////////////
#define GEMM_TPARAM_MACRO(devc) \
if (devc == 'g') { /* Small device */ \
if (devc == 'g' || devc == 'p') { /* Small device */ \
if (!transpose_a && transpose_b) { /* nt */ \
bm = 64; \
bn = 32; \