From 992558dc493676ee974f0123bd424da194b9f098 Mon Sep 17 00:00:00 2001 From: Cheng Date: Fri, 16 May 2025 20:39:47 +0900 Subject: [PATCH] Add NO_GPU_MULTI(Eig) in cuda backend --- mlx/backend/cuda/primitives.cu | 1 + 1 file changed, 1 insertion(+) diff --git a/mlx/backend/cuda/primitives.cu b/mlx/backend/cuda/primitives.cu index dc6edf606..defdc746a 100644 --- a/mlx/backend/cuda/primitives.cu +++ b/mlx/backend/cuda/primitives.cu @@ -140,6 +140,7 @@ NO_GPU(Tan) NO_GPU(Tanh) NO_GPU(Inverse) NO_GPU(Cholesky) +NO_GPU_MULTI(Eig) NO_GPU_MULTI(Eigh) namespace fast {