removed kernel hashmap

jafioti · Jan 12, 2024 · ab9df3d · ab9df3d
1 parent c506d1e
commit ab9df3d
Show file tree

Hide file tree

Showing 5 changed files with 63 additions and 217 deletions.
diff --git a/src/compilers/metal/fp16/matmul.rs b/src/compilers/metal/fp16/matmul.rs
@@ -392,7 +392,6 @@ impl Compiler for MetalMatMulCompiler {
                         dev.clone(),
                         queue.clone(),
                         &mut HashMap::new(),
-                        &graph.dyn_map,
                     ))
                     .input(src2, 0, src2_shape)
                     .finish();
@@ -496,7 +495,6 @@ impl Compiler for MetalMatMulCompiler {
                         src1_shape,
                         dev.clone(),
                         queue.clone(),
-                        &mut HashMap::new(),
                         &graph.dyn_map,
                     ))
                     .input(src1, 0, src1_shape)
@@ -510,7 +508,6 @@ impl Compiler for MetalMatMulCompiler {
                         src2_shape,
                         dev.clone(),
                         queue.clone(),
-                        &mut HashMap::new(),
                         &graph.dyn_map,
                     ))
                     .input(src2, 0, src2_shape)

diff --git a/src/compilers/metal/fp16/rms_norm.rs b/src/compilers/metal/fp16/rms_norm.rs
@@ -250,7 +250,6 @@ impl Compiler for RMSNormCompiler {
                         sh,
                         dev.clone(),
                         queue.clone(),
-                        &mut HashMap::new(),
                         &graph.dyn_map,
                     ))
                     .input(x, 0, sh)

diff --git a/src/compilers/metal/fp32/matmul.rs b/src/compilers/metal/fp32/matmul.rs
@@ -355,7 +355,6 @@ impl Compiler for MetalMatMulCompiler {
                         src1_shape,
                         dev.clone(),
                         queue.clone(),
-                        &mut HashMap::default(),
                         &graph.dyn_map,
                     ))
                     .input(src1, 0, src1_shape)
@@ -368,7 +367,6 @@ impl Compiler for MetalMatMulCompiler {
                         src2_shape,
                         dev.clone(),
                         queue.clone(),
-                        &mut HashMap::default(),
                         &graph.dyn_map,
                     ))
                     .input(src2, 0, src2_shape)

diff --git a/src/compilers/metal/mod.rs b/src/compilers/metal/mod.rs
@@ -23,7 +23,7 @@ use metal_rs::*;
 use crate::{
     op::InputTensor,
     prelude::{
-        symbolic::{BigExpression, Expression, Term},
+        symbolic::{BigExpression, Term},
         *,
     },
 };