[PIR] fix onednn double reg (#60720)

* fix onednn double reg
PaddlePaddle · Jan 11, 2024 · a1ef22c · a1ef22c
1 parent bcefbaf
commit a1ef22c
Showing 1 changed file with 0 additions and 6 deletions.
diff --git a/paddle/fluid/pir/dialect/kernel/ir/kernel_dialect.cc b/paddle/fluid/pir/dialect/kernel/ir/kernel_dialect.cc
@@ -136,9 +136,7 @@ CustomKernelDialect::CustomKernelDialect(pir::IrContext *context)
 }
 
 void CustomKernelDialect::initialize() {
-  RegisterTypes<paddle::dialect::AllocatedDenseTensorType>();
   RegisterOps<dialect::CustomKernelOp>();
-  RegisterAttributes<paddle::dialect::KernelAttribute>();
 }
 
 void CustomKernelDialect::PrintType(pir::Type type, std::ostream &os) const {
@@ -176,13 +174,9 @@ OneDNNKernelDialect::OneDNNKernelDialect(pir::IrContext *context)
 }
 
 void OneDNNKernelDialect::initialize() {
-  RegisterTypes<paddle::dialect::AllocatedDenseTensorType,
-                paddle::dialect::AllocatedSelectedRowsType,
-                paddle::dialect::AllocatedDenseTensorArrayType>();
   RegisterOps<dialect::OneDNNPhiKernelOp,
               dialect::OneDNNMixedPhiKernelOp,
               dialect::OneDNNLegacyKernelOp>();
-  RegisterAttributes<paddle::dialect::KernelAttribute>();
 }
 
 void OneDNNKernelDialect::PrintType(pir::Type type, std::ostream &os) const {