diff --git a/paddle/pten/core/infermeta_utils.cc b/paddle/pten/core/infermeta_utils.cc index 326f9aa720be23611f4f99668c96ffb55d0da552..da2f0c92aa546d810e059875d24b61bc76f80486 100644 --- a/paddle/pten/core/infermeta_utils.cc +++ b/paddle/pten/core/infermeta_utils.cc @@ -83,8 +83,8 @@ MetaTensor* InferMetaContext::MutableOutputAt(size_t idx) { return outputs_.at(idx).get(); } -MetaFunctionMap& MetaFunctionMap::Instance() { - static MetaFunctionMap g_meta_fn_map; +MetaFnFactory& MetaFnFactory::Instance() { + static MetaFnFactory g_meta_fn_map; return g_meta_fn_map; } diff --git a/paddle/pten/core/infermeta_utils.h b/paddle/pten/core/infermeta_utils.h index aed1fc2b77ff0cc8f40e8a025f910311b1016036..0c95b204abf38051077179e4cc8d7fa1c67301ae 100644 --- a/paddle/pten/core/infermeta_utils.h +++ b/paddle/pten/core/infermeta_utils.h @@ -194,9 +194,9 @@ struct InferMetaFnImpl { }; }; -class MetaFunctionMap { +class MetaFnFactory { public: - static MetaFunctionMap& Instance(); + static MetaFnFactory& Instance(); bool Contains(const std::string& kernel_name_prefix) const { return meta_fn_map_.count(kernel_name_prefix) > 0; @@ -225,7 +225,7 @@ class MetaFunctionMap { } private: - MetaFunctionMap() = default; + MetaFnFactory() = default; /** * [ Why use kernel name prefix? ] @@ -243,14 +243,14 @@ class MetaFunctionMap { */ paddle::flat_hash_map meta_fn_map_; - DISABLE_COPY_AND_ASSIGN(MetaFunctionMap); + DISABLE_COPY_AND_ASSIGN(MetaFnFactory); }; struct InferMetaFnRegistrar { InferMetaFnRegistrar(const char* kernel_name_prefix, InferMetaFn infer_meta_fn) { - MetaFunctionMap::Instance().Insert(kernel_name_prefix, - std::move(infer_meta_fn)); + MetaFnFactory::Instance().Insert(kernel_name_prefix, + std::move(infer_meta_fn)); } }; diff --git a/paddle/pten/tests/core/test_meta_fn_utils.cc b/paddle/pten/tests/core/test_meta_fn_utils.cc index c7d092c468fae3a28eb6de90e020a31e98a293d1..3cde1cfb5dcd56384776c094bb7209bb3320f95d 100644 --- a/paddle/pten/tests/core/test_meta_fn_utils.cc +++ b/paddle/pten/tests/core/test_meta_fn_utils.cc @@ -22,7 +22,7 @@ limitations under the License. */ namespace pten { namespace tests { -TEST(MetaFunctionMap, InferMetaFnExists) { +TEST(MetaFnFactory, InferMetaFnExists) { pten::DenseTensor dense_x; dense_x.Resize(pten::framework::make_ddim({3, 4})); @@ -38,7 +38,7 @@ TEST(MetaFunctionMap, InferMetaFnExists) { ctx.EmplaceBackInput(shared_meat_x); ctx.EmplaceBackOutput(shared_meta_out); ctx.SetMetaConfig(/*is_runtime=*/true); - pten::MetaFunctionMap::Instance().Get("sign")(&ctx); + pten::MetaFnFactory::Instance().Get("sign")(&ctx); EXPECT_EQ(dense_out1.dims().size(), dense_out2.dims().size()); EXPECT_EQ(dense_out1.dims()[0], dense_out2.dims()[0]);