Fix metal exports (#3081)

Author: ivarflakstad

candle-metal-kernels/src/kernels/mod.rs

@@ -24,7 +24,7 @@ pub use mlx_gemm::{call_mlx_gemm, GemmDType};
 pub use quantized::{call_quantized_matmul_mm_t, call_quantized_matmul_mv_t, GgmlDType};
 pub use random::*;
 pub use reduce::*;
-pub use sdpa::{call_sdpa_full, call_sdpa_vector, call_sdpa_vector_2pass};
+pub use sdpa::{call_sdpa_full, call_sdpa_vector, call_sdpa_vector_2pass, SdpaDType};
 pub use sort::{call_arg_sort, call_mlx_arg_sort};
 pub use ternary::call_where_cond_strided;
 pub use unary::*;

candle-metal-kernels/src/lib.rs

@@ -9,8 +9,8 @@ pub use err::MetalKernelError;
 pub use kernel::Kernels;
 pub use kernels::{
     affine::*, call_binary_contiguous, call_binary_strided, call_mlx_gemm, cast::*, convolution::*,
-    fill::*, indexing::*, quantized::*, random::*, reduce::*, sort::*, ternary::*, unary, unary::*,
-    GemmDType, GgmlDType,
+    fill::*, indexing::*, quantized::*, random::*, reduce::*, sdpa::*, sort::*, ternary::*, unary,
+    unary::*, GemmDType, GgmlDType,
 };
 use metal::{
     BlitCommandEncoder, Buffer, CommandQueue, ComputeCommandEncoder, ComputePipeline,