Remove ccv_nnc_mfa_cmul kernel, move that to CMulKernel.

liuliu · Oct 28, 2024 · e4d13c2 · e4d13c2
1 parent 34d496a
commit e4d13c2
Show file tree

Hide file tree

Showing 10 changed files with 445 additions and 313 deletions.
diff --git a/lib/nnc/cmd/pad/mps/ccv_nnc_pad_mps.m b/lib/nnc/cmd/pad/mps/ccv_nnc_pad_mps.m
@@ -73,7 +73,6 @@ static int _ccv_nnc_pad_back(const ccv_nnc_cmd_t cmd, const ccv_nnc_hint_t hint,
 		const int* const end = cmd.info.pad.end;
 		const int g_nd = ccv_nnc_tensor_nd(g->info.dim);
 		assert(g_nd == ccv_nnc_tensor_nd(a->info.dim));
-		const int type = cmd.info.pad.type;
 		MPSGraphExecutable* executable = ccv_nnc_mps_graph_executable_cache(a_key, indices, ^void (MPSGraph* graph, NSMutableArray<MPSGraphTensor*>* inputTensors, NSMutableArray<MPSGraphShapedType*>* inputShapedTypes, NSMutableArray<MPSGraphTensor*>* resultTensors) {
 			MPSGraphTensor* mps_input_g;
 			MPSGraphTensor* mps_g = ccv_nnc_mps_graph_tensor_input(graph, g, g->info.dim, g->stride, &mps_input_g);

diff --git a/lib/nnc/mfa/ccv_nnc_mfa.cpp b/lib/nnc/mfa/ccv_nnc_mfa.cpp
@@ -108,12 +108,6 @@ void mfa::cache<mfa::adam::hash, mfa::adam::pipeline>::prepare(mfa::context* con
   _mfa_cache_prepare(&map, context, hash);
 }
 
-template <>
-void mfa::cache<mfa::cmul::hash, mfa::cmul::pipeline>::prepare(mfa::context* context, mfa::cmul::hash hash)
-{
-  _mfa_cache_prepare(&map, context, hash);
-}
-
 template <>
 void mfa::cache<mfa::gemv::hash, mfa::gemv::pipeline>::prepare(mfa::context* context, mfa::gemv::hash hash)
 {

diff --git a/lib/nnc/mfa/ccv_nnc_mfa.hpp b/lib/nnc/mfa/ccv_nnc_mfa.hpp
@@ -52,7 +52,6 @@ class context {
   cache<normalization::hash, normalization::pipeline> normalization_cache;
   cache<depalettize::hash, depalettize::pipeline> depalettize_cache;
   cache<adam::hash, adam::pipeline> adam_cache;
-  cache<cmul::hash, cmul::pipeline> cmul_cache;
   cache<gemv::hash, gemv::pipeline> gemv_cache;
   cache<cast::hash, cast::pipeline> cast_cache;
   cache<add::hash, add::pipeline> add_cache;