organization

jafioti · Jan 9, 2024 · e1cf44a · e1cf44a
1 parent 67366e1
commit e1cf44a
Show file tree

Hide file tree

Showing 3 changed files with 587 additions and 617 deletions.
diff --git a/src/compilers/metal/fp16/mod.rs b/src/compilers/metal/fp16/mod.rs
@@ -2,7 +2,6 @@ use half::f16;
 
 mod matmul;
 mod mean_reduce;
-pub mod other;
 mod rms_norm;
 
 pub type MetalFp16Compiler = (
@@ -13,7 +12,10 @@ pub type MetalFp16Compiler = (
         super::other::ARangeCompiler<f16>,
         super::binary::MetalGatherCompiler<f16>,
     ),
-    (other::MetalExpCompiler, other::MetalSwishCompiler),
+    (
+        super::other::MetalExpCompiler<f16>,
+        super::other::MetalSwishCompiler<f16>,
+    ),
     matmul::MetalMatMulCompiler,
     mean_reduce::MeanReduceCompiler,
     rms_norm::RMSNormCompiler,