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

Add NO_GPU_MULTI(Eig) in cuda backend

Browse Source
This commit is contained in:
Cheng 2025-05-16 20:39:47 +09:00
parent 5e71f2f3ef
commit 992558dc49
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
mlx/backend/cuda/primitives.cu
View File

@ -140,6 +140,7 @@ NO_GPU(Tan)
NO_GPU(Tanh) NO_GPU(Tanh)
NO_GPU(Inverse) NO_GPU(Inverse)
NO_GPU(Cholesky) NO_GPU(Cholesky)
NO_GPU_MULTI(Eig)
NO_GPU_MULTI(Eigh) NO_GPU_MULTI(Eigh)
namespace fast { namespace fast {