[CustomDevice] Allow registration of not ready kernels (#57038)

ronny1996 · web-flow · commit 12301bc5337f · 2023-09-07T10:29:15.000+08:00
diff --git a/paddle/phi/core/custom_kernel.cc b/paddle/phi/core/custom_kernel.cc
@@ -16,9 +16,6 @@
 
 #include "glog/logging.h"
 
-static std::vector<std::string> gpu_exclusive_kernels({"sync_batch_norm",
-                                                       "sync_batch_norm_grad"});
-
 namespace phi {
 
 void CustomKernelMap::RegisterCustomKernel(const std::string& name,
@@ -44,16 +41,6 @@ void CustomKernelMap::RegisterCustomKernels() {
   }
   auto& kernels = KernelFactory::Instance().kernels();
   for (auto& pair : kernels_) {
-    if (kernels.find(pair.first) == kernels.cend()) {
-      if (std::find(gpu_exclusive_kernels.cbegin(),
-                    gpu_exclusive_kernels.cend(),
-                    pair.first) == gpu_exclusive_kernels.cend()) {
-        PADDLE_THROW(phi::errors::InvalidArgument(
-            "The kernel %s is not ready for custom kernel registering.",
-            pair.first));
-      }
-    }
-
     for (auto& info_pair : pair.second) {
       PADDLE_ENFORCE_EQ(
           kernels[pair.first].find(info_pair.first),
